对Pytorch模型进行了99%的验证精度训练,但在预测过程中只能预测一个类别

问题描述

我开始从facenet-pytorch库构建端到端脸部识别系统,但预测部分出现了问题。在训练过程以99%的验证精度完成后,该模型将训练数据本身错误分类,并且完全偏向一个类别(Garry_Marshall)。无论您拍摄什么图像,它仍然只会给您该类。下面提供了代码(字符串)链接,如果需要任何其他信息,请询问。有关代码位于“人脸分类标题下。

一个副问题:在训练的初始阶段,我的训练准确性将低于我的验证准确性。那怎么可能?

https://colab.research.google.com/drive/1z1ARMkXBariUOj0A-PNcrliup8kmGtpH?usp=sharing

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)