在运行yolov5时这里报错:The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3
这是由于你使用的yolov5版本是5.0但是你的pt文件不是5.0的,两者要对应。
下载
https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt
替换代码中原来的yolov5s.pt就好了
在运行yolov5时这里报错:The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3
这是由于你使用的yolov5版本是5.0但是你的pt文件不是5.0的,两者要对应。
下载
https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt
替换代码中原来的yolov5s.pt就好了