search

sfzhang15 / RefineDet

Single-Shot Refinement Neural Network for Object Detection, CVPR, 2018
Other
1.43k stars 393 forks source link

How can i train using more faster CNN like resnet10 and mobilenet? #160

Closed campos537 closed 5 years ago

campos537 commented 5 years ago

I liked a lot the results you had, but i want to train using this other CNN that are faster, to run on CPU

sfzhang15 commented 5 years ago

@campos537 You can replace Res101 (L408 and L465) with new backbones and modify L228, L253-L254 and L298 correspondingly.

campos537 commented 5 years ago

Thanks for the help!

wait1988 commented 5 years ago

Can you share your script to extract the pretrained COCO model of given classes? @sfzhang15

sfzhang15 commented 5 years ago

@wait1988 Please refer to convert_model_320.ipynb and convert_model_512.ipynb.