Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

老哥,你这个v2的速度如何? #37

Open LDoubleZhi opened 5 years ago

LDoubleZhi commented 5 years ago

fps是多少?占显存多少?我用上一个版本的代码,conf threshold也设了0.2,精度确实高,但是fps差不多10帧,显存占了10g有点太多了。

Stinky-Tofu commented 5 years ago

@LDoubleZhi 你好,目前只追求了性能,没有追求速度,所以有很多冗余代码,比较占显存,10fps不可能吧,我这边用v1的代码来测试5000张图片,conf threshold设置为0.2的话,3、4分钟就跑完了,当然模型压缩是我下一步要做的。v2指的是用darknet53.weights初始化,然后在VOC数据集上训练的,v2主要是模块化了代码,方便我后面使用轻量化的backbone。