search

tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 486 forks source link

FCOS refactor #241

Closed kfxw closed 4 years ago

kfxw commented 4 years ago

I've reimplemented the FCOS model. The ResNet-50 based one has the same performance on coco minival2014 set with mAP of 36.6, which behaves similar with the one trained with official pytorch code. Besides, the training speed is still kind of slower than official code. More refinement may be needed in the future.

kfxw commented 4 years ago

Fixed most of problems mentioned in the last review, except the 'throwout_param', which is used as input for custom op. Custom op cannot accept class type as input. Also the training is further speedup, with quite compatible running speed with the pytorch counterpart. Memory usage is ~300MB larger.