我正在尝试使用自定义图像数据来训练和测试pytorch神经网络。我有一个图像文件夹作为数据,另一个文件夹图像作为地面真理。文件和地面真相映像具有相同的名称,但它们位于单独的文件夹中。我想将它们用作图像和地面实况,例如,将1.jpg馈入网络,htis图片的结果应与其他1.jpg(地面实况)进行比较。我希望神经网络输出是图像而不是分类(例如图像到图像的翻译或pix2pix)。
我尝试过pytorch数据加载器和其他一些文档,但是它们不是我所需要的。我尝试过的一些文档:
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