yolov3-pytorch检测物体不对应

yolov3-pytorch目标检测类别不对应

目录

1.遇到的问题

2.解决方法

3.yolov3-pytorch目标检测参考的文章

---

1.遇到的问题

在按照网上的教程下载coco数据集，下载yolov3源码，修改参数，训练和测试完成后，在检测的过程中却发现在识别的时候目标对应不上，如下图将猫识别成了truck

2.解决方法

我发现我下载的detect.py里有一个--names参数，在运行时‘python3 detect.py'时需要指定相应的voc.names，否则会识别目标错乱，如下我的代码是

python3 detect.py --cfg cfg/你的.cfg文件 --weights weights/last.pt --names 你写好的.names文件 --source data/samples

其中，--cfg后面跟的是你的.cfg文件的路径，--weight跟的是你训练完生成的last.pt文件，--name则是你的.name文件的路径（如果缺少了这个参数你的识别就会错乱），--source是检测的图片的路径。

这样设置之后就可以成功检测了