我正在尝试使用自定义图像数据来训练和测试 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