yolov5+sort 修改weights后报错

Question

克隆的yolov5+sort的git项目

链接在这里

跑通代码后，将代码换成自己之前在yolov5训练的自己数据集的权重，然后就开始报错了

Answer 1

参考GPT和自己的思路，这个错误看起来是在导入模型时出现了问题。最可能的原因是模型的权重文件与模型定义不匹配，或者是模型定义本身出现了问题。以下是可能的解决方案：

1 确保修改后的权重文件与模型定义的层数匹配。如果修改了模型结构，确保修改后的模型与修改后的权重文件相匹配。

2 检查models/yolo.py文件，确保其中的代码没有错误。可以尝试恢复原始版本的yolo.py文件，看看是否能够解决问题。

3 确保使用的PyTorch版本与yolov5_sort-master要求的PyTorch版本相匹配。根据报错信息，看起来您正在使用PyTorch 1.13.1+cu116，但是这个版本号在PyTorch官方文档中找不到。可能需要升级或降级PyTorch版本以解决问题。

4 如果上述解决方案都不起作用，可以尝试重新安装yolov5_sort-master并重新构建整个项目。