栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

使用pytorch顺序读取文件夹中图像

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

使用pytorch顺序读取文件夹中图像

目的: 在使用pytorch进行神经网络训练的时候,需要每次向网络中输入一组图片,但是每次使用pytorch内置的ImageFolder,DatasetFolder等函数读取文件夹中的文件时,里面的文件总是不是按照顺序来读取的。
它的排序方法是:1.tif,10.tif,100.tif…我们希望他是按照顺序来读取,所以有了以下的解决方法:

# 文件夹路径
path = "./splitdata/splitdata"
dir_list = os.listdir(path)

# 将顺序读取的文件保存到该list中
origin_data = []
for i in range(len(dir_list)):
    m = i+1
    tpath = os.path.join(path,str(m)+".tif")
    img = io.imread(tpath)
    transform = transforms.Compose([transforms.ToTensor()])
    data = transform(img)
    origin_data.append(data) # size:10807
dataloader_original = DataLoader(origin_data, batch_size=Batch_Size, shuffle=False)
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/923891.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号