Thanks for your reply. One root contains the original images and the other contains the corresponding ground truth, which are also images (not a classification label like “dog”). So,  there is some difficulty to combine these two folders together in one dataloader.