人脸表情识别数据集的正确使用方法(Affectnet、RAF-DB、FERPlus数据集通用)
众所周知深度学习是个很玄学的东西,不同的数据集加载方式会训练出不一样的精度,导致无法复现论文精度。 这里分享下正确的加载数据集的方法:
拿RAF-DB数据集举例:
①准备好RAF-DB数据集,训练集和测试集放进同一目录,标签也都放进同一文件(list_patition_label.txt)。目录结构如下:
②把对应的dataset文件放进目录,更改标签文件路径:
③在train文件中使用dataset加载数据集即可:
FERPlus和AffectNet也是类似的使用方法,亲测可以成功复现论文精度,复现不出精度的可以用这种方法加载数据集试一试。