Yet-Another-EfficientDet-PytorchPyTorch 版本的 EfficientDet
去年 11 月份，谷歌大腦提出兼顧準(zhǔn)確率和模型效率的新型目標(biāo)檢測(cè)器 EfficientDet，實(shí)現(xiàn)了新的 SOTA 結(jié)果。
Yet-Another-EfficientDet-Pytorch 是具有 SOTA 實(shí)時(shí)性能的官方 EfficientDet 的 pytorch 重新實(shí)現(xiàn)。該版本的性能接近原版，但速度是官方 TensorFlow 實(shí)現(xiàn)的近 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 |
評(píng)論
圖片
表情