YOLO模型train.py文件报错
问题
在训练YOLO模型后将自己权重用于预测时总是提示如下报错信息：
RuntimeError: Error(s) in loading state_dict for YoloBody:
size mismatch for last_layer0.6.weight: copying a param with shape torch.Size([24, 1024, 1, 1]) from checkpoint, the shape in current model is torch.Size([27, 1024, 1, 1]).
。。。。。。。。
此处省略
解决办法:
将程序中 config.py下的class数改为 需要预测的类别数+1，比如需要预测猫与狗，则class=3，因为需要加上背景，否则在预测时会出现错误，导致白白训练。
文章出处登录后可见！
已经登录？立即刷新
