Yet-Another-EfficientDet-PytorchPyTorch 版本的 EfficientDet
去年 11 月份,谷歌大脑提出兼顾准确率和模型效率的新型目标检测器 EfficientDet,实现了新的 SOTA 结果。
Yet-Another-EfficientDet-Pytorch 是具有 SOTA 实时性能的官方 EfficientDet 的 pytorch 重新实现。该版本的性能接近原版,但速度是官方 TensorFlow 实现的近 26 倍!
性能
Pretrained weights and benchmark
coefficient | pth_download | GPU Mem(MB) | FPS | Extreme FPS (Batchsize 32) | mAP 0.5:0.95(this repo) | mAP 0.5:0.95(paper) |
---|---|---|---|---|---|---|
D0 | efficientdet-d0.pth | 1049 | 36.20 | 163.14 | 32.6 | 33.8 |
D1 | efficientdet-d1.pth | 1159 | 29.69 | 53.82 | 38.2 | 39.6 |
D2 | efficientdet-d2.pth | 1321 | 26.50 | 40.43 | 41.5 | 43.0 |
D3 | efficientdet-d3.pth | 1647 | 22.73 | - | 44.9 | 45.8 |
D4 | efficientdet-d4.pth | 1903 | 14.75 | - | 48.1 | 49.4 |
D5 | efficientdet-d5.pth | 2255 | 7.11 | - | 49.5 | 50.7 |
D6 | efficientdet-d6.pth | 2985 | 5.30 | - | 50.1 | 51.7 |
D7 | efficientdet-d7.pth | 3819 | 3.73 | - | 50.7 | 52.2 |
Speed Test
coefficient | Time | FPS | Ratio |
---|---|---|---|
Official D0 (tf postprocess) | 0.713s | 1.40 | 1X |
Official D0 (numpy postprocess) | 0.477s | 2.09 | 1.49X |
Yet-Another-EfficientDet-D0 | 0.028s | 36.20 | 25.86X |
评论