目的: 在使用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)



