yolov5_ROS icon indicating copy to clipboard operation
yolov5_ROS copied to clipboard

你好,请问为什么用你的代码和官方的detect.py 用同一个权值文件识别的结果不一样。识别的class都是全部的

Open github-lile opened this issue 3 years ago • 3 comments

github-lile avatar Jul 15 '21 11:07 github-lile

用的哪个python文件,就修改里面的classes变量,换成你需要的识别类别。

例如:用的是final_yolo.py 文件来识别,那么: 修改93行: 我的是 classes = (0,1,2,3,5,7)

改成你需要识别的类别,例如:只看人 换成 classes =(0)

参考: yolo类别目录: https://blog.csdn.net/Dreaming_of_you/article/details/102756445

qianmin avatar Jul 17 '21 12:07 qianmin

感谢你的回复,我是把classes设置成了None,所以识别的是全部,然后进行图片的识别,结果跟官方的识别文件结果不一样,final_yolo.py会有误识别。

-----原始邮件----- 发件人:qianmin @.> 发送时间:2021-07-17 20:51:32 (星期六) 收件人: qianmin/yolov5_ROS @.> 抄送: github-lile @.>, Author @.> 主题: Re: [qianmin/yolov5_ROS] 你好,请问为什么用你的代码和官方的detect.py 用同一个权值文件识别的结果不一样。识别的class都是全部的 (#4)

用的哪个python文件,就修改里面的classes变量,换成你需要的识别类别。

例如:用的是final_yolo.py 文件来识别,那么: 修改93行: 我的是 classes = (0,1,2,3,5,7)

改成你需要识别的类别,例如:只看人 换成 classes =(0)

参考: yolo类别目录: https://blog.csdn.net/Dreaming_of_you/article/details/102756445

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

github-lile avatar Jul 17 '21 14:07 github-lile

如果是自己的训练权重,不在这80个类别里,请问在哪里修改啊?

bsljsljy avatar Sep 05 '23 08:09 bsljsljy