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.