Pytorch模型的训练验证准确率为99%,但在预测过程中仅预测一类

2024-04-26 01:05:30 发布

您现在位置:Python中文网/ 问答频道 /正文

我从facenet pytorch库开始构建端到端的人脸识别系统,但预测部分出现了问题。训练过程以99%的验证准确率完成后,模型将训练数据本身分类错误,并且完全偏向于一类(Garry_Marshall)。不管你拍的是什么样的图像,它仍然只会给你那一类。下面给出了代码(colab)链接,如果需要任何其他信息,请询问。问题代码是标题“人脸分类”下的部分

附带问题:我的培训准确度将低于培训初始阶段的验证准确度。这怎么可能

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


Tags: 数据代码模型过程系统错误分类pytorch