Pytorch загружает изображения и наземные изображения для обучения нейронной сети

Я пытаюсь использовать пользовательские данные изображения для обучения и тестирования нейронной сети pytorch. У меня есть папка с изображениями в качестве данных и другая папка с изображениями в качестве истины. Файлы и наземные изображения имеют одинаковое имя, но находятся в отдельной папке. Я хочу использовать их как изображение и наземную правду как пару, например, 1.jpg подайте в сеть, и результат htis изображения должен сравниться с другим 1.jpg (его наземная правда). Я хочу, чтобы вывод моей нейронной сети был изображением, а не классификацией (например, перевод изображения в изображение или pix2pix).

Я пробовал pytorch dataloader и некоторые другие документы, но они были не тем, что мне было нужно. Некоторые документы, которые я пробовал:

https://www.kaggle.com/leighplt/pytorch-tutorial-dataset-data-preparetion-stage

https://towardsdatascience.com/building-efficient-custom-datasets-in-pytorch-2563b946fd9f

https://pytorch.org/tutorials/beginner/data_loading_tutorial.html