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 这是官方的模型啊,你应该要用你自己的模型才对。
【相关推荐】
首先还是打开根目录下的train.py,直接看parse_opt()
已经提示你了,yaml文件和pt文件没对应上,yaml文件就是你训练时候用的yaml,pt文件是你训练出来的结果bast.pt,而不是用你训练时候的预训练模型pt