ZBigFish / FER-VT

The unofficial implementation of paper "Facial expression recognition with grid-wise attention and visual transformer"
16 stars 2 forks source link

CK+与RAF-DB数据集 #1

Open CHENSY-SIYU opened 2 years ago

CHENSY-SIYU commented 2 years ago

请问您的CK+数据集和RAF数据集是怎么处理的,我并没有在dataset的文件夹中看到对应的结构

ZBigFish commented 2 years ago

CK+的数据结构是按照ImageFolder来存放的,rafdb同理,其中CK的数据集读取在train里面有对应方法。raf的数据集读取我忘记上传了,主要是忘了放哪了,你可以自行找一个。当然,我的代码风格比较乱,你可以自行写一个数据集的读取。

ZBigFish commented 2 years ago

CK+的数据结构是按照ImageFolder来存放的,rafdb同理,其中CK的数据集读取在train里面有对应方法。raf的数据集读取我忘记上传了,主要是忘了放哪了,你可以自行找一个。当然,我的代码风格比较乱,你可以自行写一个数据集的读取。