Need help with Thundernet_Pytorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

ouyanghuiyu
253 Stars 68 Forks 49 Commits 36 Opened issues

Description

Implementation Thundernet

Services available

!
?

Need anything else?

Contributors list

No Data

Thundernet_Pytorch

项目已经更新:mmdetection_lite

20191222 update

  • add data augmentation
  • add Multi-scale training
  • add onnx (doing)

pretrained model

  • train code in : https://github.com/ouyanghuiyu/Snet

Requirements

  • pytorch 1.2.0
  • torchvision 0.4

Lib Prepare

git clone https://github.com/ouyanghuiyu/Thundernet_Pytorch.git

Build

cd lib && python setup.py  build_ext --inplace
cd psroialign/PSROIAlign && sh build.sh 

Data Prepare

Download VOC0712 datasets ln -s "YOUR PATH" data

Train

cd script
sh  train_49.sh
sh  train_146.sh
sh  train_535.sh

demo

cd script
sh  pre.sh

TODO LIST

  • add coco train and test
  • add NCNN inference

Citation

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}
}

VOC TEST EXAMPLE

test

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.