- 用pytorch实现yolov3算法不是特别合适,目前网上的一些yolov3 pytorch复现经过我的亲身经历,感觉都有些问题。主要是test.py中的evaluate函数在yolo检测不到框的时候就会报错,我看了github上源代码的issues,认为解决方法就是如果报错就跳过evaluate,但我自己没有试过,以后有机会再添这坑。
- 目前我用过的比较好的yolov3复现是在这里:
git clone https://github.com/pjreddie/darknet
可以参考这两个博客
https://blog.youkuaiyun.com/qq_35451572/article/details/80384674
https://blog.youkuaiyun.com/lilai619/article/details/79695109
里面讲的比较详细了,我这就对其中一些小坑进行总计。
3. 自己上手不要用COCO VOC这些数据集,训练要很久才有效果。
4. 训练中出现nan是正常的,前期检测不到就会出现nan的
5. 如果要用gpu要装cuda和cudn,并且配置好。
6. 训练或者测试前要注意cfg有没有修改好相应的batchsize。
7. make -j 出现警告C4996 ‘cudaThreadSynchronize’: 被声明为已否决
参考这个:
https://blog.youkuaiyun.com/qq_42403190/article/details/90109282