- Python 3.8.5 x64
- Pytorch
- Pandas
- Bypy(可选)
- numpy
- PyCharm 2021.03
这一次使用到了ResNet50, DenseNet121和MobileNetV2三个模型进行训练,在程序文件中,其中:
- DenseNetModel.py
- ResNetModel.py
- MobileNetModel.py 为模型文件,在训练测试时,运行其中的:
- DenseNetTrain.py
- ResNetTrain.py
- MobileNetTrain.py 即可训练该模型。在训练过程中,可选择是训练”性别(输入G)“或是”表情(输入E)“数据集: 输入训练的次数: 输入训练结果保存为csv时的给自己看的参数(会出现在该csv文件的命名里面):
每一次训练完成之后,其训练过程中的loss和accuracy会以CSV文件形式全部保存在/DataSave/CSV目录下。
同时为了方便服务器训练,也加入了可以将文件同步至百度网盘(主要是为了方便后面的调参),假如有需要用到可以如下操作:
pip install bypy==1.6.10
输入
bypy info
在每一次训练完成之后,其训练完成的CSV文件会自动同步至自己的百度网盘:
直接运行在/DataSave/CSV目录下”CSV2JPG"文件即可,所载入的csv名称需要自己修改一下。
性别识别,30Epoch,最高准确率为86.7%; 表情识别,30Epoch,最高准确率为88.9%;
性别识别,30Epoch,最高准确率为88.5%; 表情识别,30Epoch,最高准确率为90.0%;
发现其中还有问题,等修改完再来画图。
- 最近时间有点赶(计算机控制作业要先搞定),可能要过一段时间才能继续完善,最新的进度都会更新到这个github repo;
- 修正一下MobileNet模型;
- 完成在服务器端直接绘图并同步至百度网盘;
- Vision Transformer的进一步完善;
- MindSpore和PeddlePeddle的粗略使用;