Solved by ptrblck in post #2 
              
              
                You could create your own Dataset like this: 
class MyDataset(Dataset):
    def __init__(self, data_paths):
        self.paths = data_paths 

    def __getitem(self, index):
        image, target_image = load_images(self.paths[index]) # load as np.array or PIL.Image
        # transform your images h…