EfficientDet family of models based on EfficientDet paper.
Main features of EfficientDet models are EfficientNet backbone and BiFPN neck.
Backbone | Head | Style | Lr schd | Mem (GB) | Inf time (fps) | box AP | Download |
---|---|---|---|---|---|---|---|
efficientnet_b0 | retinahead | pytorch | 300e | 14.6 | 29.3 | 33.9 | model |
Note:
- retinahead_effd0_bifpn_1x_coco is not an official model in the EfficientDet paper. Everything but the head is similar to the model in the paper.
- retinahead_effd0_bifpn_1x_coco was trained on 8 Tesla V100 cards