Open zhangqianhui opened 7 years ago
If you see ./src/util.lua
, prepImageNet
function simply goes to ImageNet directory and count and read all file names. Just change line 70~80 in ./train-SRResNet.lua
into CelebA or something. Fix prepImageNet
as you wish because the dataset directory structure is likely to be different.
thanks
Imagenet is very big and How to use celebA or other dataset to train SRGAN? Can you help me ?