- pytorch 1.2.0
- torchvision 0.4
- add focal loss
- add more data aug
git clone https://github.com/ouyanghuiyu/Thundernet_Pytorch.git
cd versions/psroialign/PSROIAlign && sh build.sh
Download VOC0712 datasets Then cd datapre run voc2txt.py (I have do it )
python3 Train_thundenet.py
python3 Testing.py
- add voc test result
- add coco train and test
- add NCNN inference
Please cite the paper in your publications if it helps your research:
@article{zheng2019thundernet,
title={ThunderNet: Towards Real-time Generic Object Detection},
author={Zheng Qin, Zeming Li,Zhaoning Zhang,Yiping Bao,Gang Yu, Yuxing Peng, Jian Sun},
journal={arXiv preprint arXiv:1903.11752},
year={2019}
}