search

WuJie1010 / Facial-Expression-Recognition.Pytorch

A CNN based pytorch implementation on facial expression recognition (FER2013 and CK+), achieving 73.112% (state-of-the-art) in FER2013 and 94.64% in CK+ dataset
MIT License
1.78k stars 548 forks source link

有没有考虑过python3版本？ #91

Open RuiqingGao opened 4 years ago

ZangHuanyu commented 4 years ago

确实有不少要改的

WuJie1010 commented 3 years ago

您可以用py3的pytorch，改动基本不大，只有极少的py版本的内部改动。

xurui123456 commented 1 year ago

您可以用py3的pytorch，改动基本不大，只有极少的py版本的内部改动。

img = img[:, :, np.newaxis] img = np.concatenate((img, img, img), axis=2) img = Image.fromarray(img)

print(type(img))

if self.transform is not None: img = self.transform(img) fer文件里的总是报错TypeError: pic should be PIL Image or ndarray. Got ＜class ‘torch.Tensor‘＞应该怎么改啊