neural-network - 用于训练神经网络的 Pytorch 加载图像和地面实况图像

Question

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

标签： neural-networkpytorchtorchvision