VGG-16算法-Pytorch实现人脸识别

>- **🍨 本文为[🔗365天深度学习训练营](https://mp.weixin.qq.com/s/0dvHCaOoFnW8SCp3JpzKxg) 中的学习记录博客**
>- **🍖 原作者：[K同学啊](https://mtyjkh.blog.csdn.net/)**

一、 前期准备

运行结果

二、调用官方的VGG-16模型

运行结果

三、 训练模型

运行结果

测试集准确率较低，约为20％，以下对优化器和初始学习率进行调整

1.将初始学习率从0.0001调整为0.001

结果为30％左右

2.将初始学习率从0.0001调整为0.01；并将优化器修改为Adam

运行结果为

3.将初始学习率从0.0001调整为0.001；并将优化器修改为Adam，训练次数改为50

结果

四、 结果可视化

出现了不显示图片预测结果的问题，对代码进行调整

结果为

五、总结

最终准确率也只达到了50％左右，没有达到预期，初始学习率的设置和优化器的使用，梯度下降的速率，都会影响最终的结果，进行下一步改进可以采取改变VGG框架的方式。