Using Pytorch to sequentially read pictures(使用pytorch进行图像的顺序读取)。

最新推荐文章于 2024-07-15 08:29:43 发布

头发结冰海鲜过敏的鱼

最新推荐文章于 2024-07-15 08:29:43 发布

阅读量6.8k

点赞数 2

分类专栏：人工智能文章标签：图像处理 pytorch

本文链接：https://blog.csdn.net/whyymlm/article/details/77365847

版权

本文介绍了在使用Pytorch训练神经网络时，如何按顺序读取两个文件夹中的配对图像。作者通过三个测试实验，分别尝试了使用dataloader的迭代器、直接排序文件名和自定义访问文件名的方法。实验结果显示，直接构造访问文件名的方式能够成功实现顺序读取图像。

摘要由CSDN通过智能技术生成

使用pytorch进行图像的顺序读取。

产生此次实验的原因：当我使用pytorch进行神经网络的训练时，需要每次向CNN传入一组图像，并且这些图片的存放位置是在两个文件夹中：

A文件夹：图片1a，图片2a，图片3a……图片1000a

B文件夹：图片1b, 图片2b，图片3b……图片1000b

所以在每个循环里，我都希望能从A中取出图片Na,同时从B文件夹中取出对应的图片Nb。

测试一：通过pytorch官方文档中的dataloader搭配python中的迭代器iterator

dataset = dset.ImageFolder(  
    root='./folder1',  
    transform=transforms.Compose([  
            transforms.ToTensor(),  
            transforms.Normalize((0.5,0.5,0.5), (0.5,0.5,0.5)), # bring images to (-1,1)  
        ])  
)  
dataloader1 = torch.utils.data.DataLoader(dataset, batch_size=opt.batchSize,   shuffle=True, num_workers=opt.workers)  
iterator1 = iter(dataloader1)
for i in range(1,1001):