xinntao / SFTGAN

CVPR18 - Recovering Realistic Texture in Image Super-resolution by Deep Spatial Feature Transform

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

the structure of training set

Marshall-yao opened this issue · comments

Thanks for sharing your codes with us!

I follow the guidance of the readme to organize the structure of the training set as follows.

It is img and bicseg files in train files .
The structure under the bicseg folder is similar to bicseg/animal/animal_segprob/.pth files.
The structure under the img folder is similar to img/grass/grass_colorimg/.png images and
Img/grass/grass_byteimg/.png images.

However , there is an error when training the code.
It is No such file or directory: '/home/dataset/OST/train/bicseg/plant/plant_colorimg/plant_653.pth'.

I think the reason for the error lies in the structural organization of the training set. Because on line 58 of the LRHR_seg_bg_dataset.py file,
seg = torch.load(HR_path.replace('/img/','/bicseg/').replace('.png', '.pth')),
after replacing OST/train/img with OST/train/bicseg, there is no .png images in the bicseg file.

Should I remove the subsequent replace?

Thanks so much.