Pytorch加载图像和地面真相图像以训练神经网络

时间:2019-09-01 08:23:38

标签: neural-network pytorch torchvision

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

0 个答案:

没有答案