YOLO5上运行val.py报错

YOLO5s上运行train.py可以正确调用自己的数据集,但是运行val.py文件是时一直报错AssertionError: yolov5s.pt (80 classes) trained on different --data than what you passed (2 classes). Pass correct combination of --weights and --data that are trained together,有大佬遇到过吗?

这个报错显示应该是你调用错模型文件了,这里说调用的了yolov5s.pt 这是官方的模型啊,你应该要用你自己的模型才对。

【相关推荐】




如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^

已经提示你了,yaml文件和pt文件没对应上,yaml文件就是你训练时候用的yaml,pt文件是你训练出来的结果bast.pt,而不是用你训练时候的预训练模型pt