A Pytorch0.4 re-implementation of Fssd detection network, with pretrained weights on VOC0712 and mAP=79.74.
Network | train data | test data | mAP | FPS | Download Link |
---|---|---|---|---|---|
FSSD | VOC 07+12 | VOC 07 test | 79.74 | 90 | Baidu/Google |
Official caffe implementation is here, and pytorch0.3 re-implementation is here.
For details, please read the paper: FSSD:Feature Fusion Single Shot Multibox Detector
Our re-implementation is slightly better than paper (79.74 vs 78.8). Inference time is tested on TITAN X Pasal and cudnn V5. More high performance models will be available soon.
I add a prune file for FSSD, which bases on Network Slimming (ICCV 2017), for details, please refer to the paper: Network Slimming.
State | mAP | FPS | GPU | Model Size | Download Link |
---|---|---|---|---|---|
Original | 79.74 | 90 | TITAN X Pascal | 136M | Baidu/Google |
First Prune (50%) | 79.64 | 150 | TITAN X Pascal | 52M | Bd(p6fe)/Gg |
Second Prune |
Steps for pruning the model: