使用VGG网络做信号灯树叶遮挡的图像分类问题: 最近接到一个二分类的任务,使用VGG训练的模型,结果发现训练准确率很高测试准确率去很低。不管怎么改权重改模型都一样。 原因分析: 经过长时间的分析确定了训练的代码没问题,权重也没问题,那么结果就出在了预测代码上。最终发现是权重没有加载,等于我每次都用初始化的模型来预测的。 解决方案: 大家遇到这种问题不妨看看权重有没有加载上 net.load_state_dict(model)