Austin Johnson

用的哪个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

你好，我没有修改，使用的是yolov5 3.0原始模型。 4.0没有测试过

一看就是颜色通道的问题，cv2.cvtcolor 搜一搜，你就懂了。

final_yolo.py里面，看到xyxy变量没？那就是框的x y w h坐标，然后学一学ros怎么发布自定义话题就行了。不难的。

可以，但是要把对应的coco.names和yaml文件换成自己的。 也可以参考的我仓库里面的yolov5_industry。

> > 可以按这位老哥贴的csdn博文来改那几行官方代码，不需要转pth > > 成功了吗？我看到CSDN博文生成是先load yolov5s.pt，再保存导出的，但是我导出的和博主导出的pth不一样~ 没成功，还是不行。

> I use coco to pretrain a model with seg and track, but the loss of cost_volume is `nan.``` Well, you said you have trained this seg model. So do...

挂个梯子噢

> https://github.com/JialianW/TraDeS/blob/3eafd249ca0f18af8000d5798d4c552a0bd627ec/experiments/youtube_vis.sh Well，this sh file only shows how to use seg in coco dataset, So there is other sh file tell us how to use seg in our own video?...