作业记录 WEEK8

最新推荐文章于 2022-08-18 15:36:31 发布

Sycret

最新推荐文章于 2022-08-18 15:36:31 发布

阅读量229

点赞数

分类专栏：遮挡状态下的活体人脸身份识别

本文链接：https://blog.csdn.net/qq_34131692/article/details/115050419

版权

这篇博客记录了作者将训练好的PyTorch模型（CBAM_Face_attention_Resnet_maskV2）用于人脸识别，替换dlib的过程。内容包括模型保存的后缀名差异（.pt, .pth, .pkl）以及如何加载只保存权重的模型。在加载模型后，由于输入维度不匹配导致错误，调整输入维度后遇到类型错误，但由于算力耗尽，问题未解决。" 128204896,12528708,Unity项目与Git协同开发指南,"['Unity', '游戏开发', 'Git', '版本控制', 'GitHub']

摘要由CSDN通过智能技术生成

inference.py

把在week08训练好的模型放入week4的作业里，替代dlib人脸识别系统中的人脸识别模型，主要是把训练好的模型保存到文件中，然后单独写一个使用模型的文件inference.py

测试数据准备：

2.jpg 和 3.jpg是同一个人
1.jpg是另外一个人

先使用dlib试一下

更换模型步骤：

选择好要换的模型，我这个模型不太行，但是还是将就用一下：
model_18_triplet_epoch_12_rocNotMasked0.571_rocMasked0.567notmaskV2.pt

.pt文件保存形式：

我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件，这几种模型文件在格式上有什么区别吗？其实它们并不是在格式上有区别，只是后缀不同而已（仅此而已），在用torch.save()函数保存模型文件时，各人有不同的喜好，有些人喜欢用.pt后缀，有些人喜欢用.pth或.pkl.用相同的torch.save（）语句保存出来的模型文件没有什么不同。

在pytorch官方的文档/代码里，有用.pt的，也有用.pth的。一般惯例是使用.pth,但是官方文档里貌似.pt更多，而且官方也不是很在意固定用一种。
————————————————
版权声明：本文为CSDN博主「wanghua609」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_38145317/article/details/103582549

所以分析一下怎么加载模型主要是看当时保存的是整个模型还是只保存了参数（这个决定了要不要把网络结构加载进来）
查看保存模型代码：