有没有考虑过python3版本？

[RuiqingGao]

[ZangHuanyu]

确实有不少要改的

[WuJie1010]

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

[xurui123456]

您可以用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‘＞应该怎么改啊