the size of tensor a(48) must match the size of tensor b(84) at non-singleton dimension 3

Sharpiless / Yolov5-Deepsort

最新版本yolov5+deepsort目标检测和追踪，能够显示目标类别，支持5.0版本可训练自己数据集

GNU General Public License v3.0

969 stars 160 forks source link

Open yjdqk opened 2 years ago

yjdqk commented 2 years ago

更换自己训练的检测模型后，报错 the size of tensor a(48) must match the size of tensor b(84) at non-singleton dimension 3

hanfengfan commented 1 year ago

请问这个问题解决了吗，我用我训练的权重也会有这个问题，用yolo提供的yolov5s就没有问题，但是用yolov5x也会有这个问题已解决，训练模型时用5.0版本的yolov5