Thundernet_Pytorch

by ouyanghuiyu

ouyanghuiyu / Thundernet_Pytorch

Implementation Thundernet

235 Stars 69 Forks Last release: Not found 48 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

Thundernet_Pytorch

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.