A question about training the FPN

[JimjimChen]

First of all, thank you very much for your work. I met the above problem when I trained FPN. Do you know how to solve it？ Looking forward to your reply.

[chaofengc]

It seems that your data is incorrect. Please follow readme and generate the correct data.

Download CelebAHQ-Mask dataset. Generate CelebAMask-HQ-mask and CelebAMask-HQ-mask-color with the provided scripts in CelebAMask-HQ/face_parsing/Data_preprocessing/.

After that, put the folder CelebAMask-HQ-mask and CelebAMask-HQ-mask-color under ../datasets/CelebAMask-HQ

[JimjimChen]

It seems that your data is incorrect. Please follow readme and generate the correct data.

Download CelebAHQ-Mask dataset. Generate CelebAMask-HQ-mask and CelebAMask-HQ-mask-color with the provided scripts in CelebAMask-HQ/face_parsing/Data_preprocessing/.

After that, put the folder CelebAMask-HQ-mask and CelebAMask-HQ-mask-color under ../datasets/CelebAMask-HQ

Many thanks！I have another question： when I finished generating 'CelebAMask-HQ-mask' , is the next step to run 'g_color.py’ directly?
When I run 'g_color.py' , I met a error : NameError : name 'im_base' is not defined. Do you know how to fix it？

[chaofengc]

Actually, training FPN does not need the colored mask. Please refer to the code:

PSFRGAN/data/celebahqmask_dataset.py

Lines 28 to 29 in 82a8755

	self.img_dataset = sorted(make_dataset(os.path.join(opt.dataroot, 'CelebA-HQ-img')))
	self.mask_dataset = sorted(make_dataset(os.path.join(opt.dataroot, 'CelebAMask-HQ-mask')))

You may simply put the generated mask folder CelebAMask-HQ-mask and corresponding image folder CelebA-HQ-img under the dataroot.