WuJie1010 / Facial-Expression-Recognition.Pytorch

A CNN based pytorch implementation on facial expression recognition (FER2013 and CK+), achieving 73.112% (state-of-the-art) in FER2013 and 94.64% in CK+ dataset
MIT License
1.77k stars 546 forks source link

萌新提问 #144

Open WhiteCranes opened 5 months ago

WhiteCranes commented 5 months ago

请问 mainpro_FER.py 执行完成之后,可以这样保存模型到文件吗? torch.save(net, '80.pth')

我训练了80代,另外再次加载模型时,怎么把一张正常的人脸图片传入模型呢?目前我把一个人脸图片裁剪为48*48,但是推理结果的正确度不好

David-want-to-fly commented 5 months ago

这个模型的效果本身就不好,自己跑训练集会更好