请教CASIA数据集训练代码

[deepthman]

首先感谢作者。我在使用自定义的数据集训练模型，使用的是代码中CASIA数据集的数据存放格式。发现其中有个Au的文件夹，看代码像是存放没有ps过的图片。代码中只用了Au图片没有添加mask图片路径

后续在使用过程中会直接报越界的错误

不知道是我的用法有问题还是哪里疏忽了。