Yet-Another-EfficientDet-PytorchPyTorch 版本的 EfficientDet

联合创作 · 2023-09-26 05:33

去年 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
浏览 3
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报