YOLO5上运行val.py报错

Question

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，有大佬遇到过吗？

Answer 1

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