Yet-Another-EfficientDet-Pytorch PyTorch 版本的 Effic开源项目

我要开发同款
匿名用户2020年04月14日
40阅读
开发技术Python
所属分类人工智能、机器学习/深度学习
授权协议LGPL

作品详情

去年11月份,谷歌大脑提出兼顾准确率和模型效率的新型目标检测器EfficientDet,实现了新的SOTA结果。

Yet-Another-EfficientDet-Pytorch 是具有SOTA实时性能的官方 EfficientDet 的pytorch重新实现。该版本的性能接近原版,但速度是官方TensorFlow实现的近26倍!

性能

Pretrainedweightsandbenchmark

coefficientpth_downloadGPUMem(MB)FPSExtremeFPS(Batchsize32)mAP0.5:0.95(thisrepo)mAP0.5:0.95(paper)D0efficientdet-d0.pth104936.20163.1432.633.8D1efficientdet-d1.pth115929.6953.8238.239.6D2efficientdet-d2.pth132126.5040.4341.543.0D3efficientdet-d3.pth164722.73-44.945.8D4efficientdet-d4.pth190314.75-48.149.4D5efficientdet-d5.pth22557.11-49.550.7D6efficientdet-d6.pth29855.30-50.151.7D7efficientdet-d7.pth38193.73-50.752.2

SpeedTest

coefficientTimeFPSRatioOfficialD0(tfpostprocess)0.713s1.401XOfficialD0(numpypostprocess)0.477s2.091.49XYet-Another-EfficientDet-D00.028s36.2025.86X
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论