关于使用自己训练的.pt文件

[lazy-codebear]

请问这个代码能不能直接把里面的yolov5s.pt换成自己训练的结果之后运行。

[qianmin]

可以，但是要把对应的coco.names和yaml文件换成自己的。

也可以参考的我仓库里面的yolov5_industry。

[yuqikong]

可以，但是要把对应的coco.names和yaml文件换成自己的。

也可以参考的我仓库里面的yolov5_industry。

可是coco.names和yaml文件在哪个目录里呢？

[bsljsljy]

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

可是coco.names和yaml文件在哪个目录里呢？

大佬，你找到了吗？