yolov3报错RuntimeError: Error(s) in loading state_dict for Darknet

这是我在调 ultralytics/yolov3 时遇到的问题
环境配置好后,训练以及画训练结果图都很顺利
如题错误出现在最后的检测

官方给的运行命令是这样的

python detect.py --weights weights/latest.pt

于是报错。绝大多数解决方法说是pytorch版本问题,但是并没解决我的错误。

在运行下面这个命令后,错误神奇消失,,原理并不懂,反正问题解决了,在这记录一下。如果有懂的大佬给我指点一二再好不过啦~

python detect.py --data-cfg data/classification.data --cfg cfg/yolov3.cfg --weights weights/latest.pt

P.S. 上面这个命令中classification.data是我自己创建的文件,大家看着修改。

你可能感兴趣的:(Python)