分类模型训练完成,却预测不准的原因

        最近,在用resnet模型进行图像分类的时候,出现了个问题。就是在test时候,准确率能达到93,单单张图片预测的时候,结果几乎没有正确的。

        最开始以为模型训练的时候的问题,因为我想要最大概率的结果,而训练的时候是各类标签的概率,所以还需要加一个torch.softmax()。

        可我后来又发现,那为什么在test数据集上的准确率却很高。应该是一张图片预测的函数里出现的问题。开始排查是不是标签与类别名没有对应上,通过多次试验,不管什么图片几乎都是那一种类别,且概率都是100%。

        最后,在一个学长的帮助下,发现因为前面训练集和测试集都使用transform对图片进行处理,而单张图片是使用cv2进行处理,但是通过测试还没有解决问题。

        终于,我们发现在加载完模型权重后没有用model.eval()对模型中的batch_Normalization

和dropout进行停用,导致模型在预测的时候对一些权重丢弃,并且对数据进行了归一化,使得预测结果大幅度偏于某一种类。 

        总结,在预测时,对图片的处理一定与前面训练集保持一致,并且别忘了model.eval()这个函数。

  • 3
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值