Home

Awesome

Fssd.pytorch

A Pytorch0.4 re-implementation of Fssd detection network, with pretrained weights on VOC0712 and mAP=79.74.

Networktrain datatest datamAPFPSDownload Link
FSSDVOC 07+12VOC 07 test79.7490Baidu/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.

About more details

Update

I add a prune file for FSSD, which bases on Network Slimming (ICCV 2017), for details, please refer to the paper: Network Slimming.

StatemAPFPSGPUModel SizeDownload Link
Original79.7490TITAN X Pascal136MBaidu/Google
First Prune (50%)79.64150TITAN X Pascal52MBd(p6fe)/Gg
Second Prune

Steps for pruning the model: