Open RRdmlearning opened 2 years ago
Well, as you can see the src_face
, tgt_face
and tgt_mask
are images read by cv2, you don't need to take care of the directory structure. Just give the correct paths of src_face
, tgt_face
and tgt_mask
to cv2.imread(PATH)
should work well.
And the indexes of target image and mask should be same.
mask images can't read in this path
It seems mask image name not match the code tgt_mask = cv2.imread(os.path.join(self.mask_root, "{}.png".format(tgt_idx)))
shuoud I make mask image by myself
I have downloaded CelebAMask-HQ and CelebA-HQ Their directory structure is as follows:
CelebAMask-HQ:
It looks like this code does not match this directory structure Am I missing something? How can I set the image path during inference