pytorch读取指定train.txt、test.txt文件加载自己的数据集
txt生成脚本完整代码如下:
读取自己的数据集,打乱并划分,生成train.txt、test.txt (每一行为图片的绝对路径+标签,完整代码,注释详细)
https://blog.csdn.net/weixin_44414948/article/details/110205546
train.txt、test.txt示例如下图所示:
pytorch数据集加载完整代码:
import torch
from torchvision import transforms
from torch.utils.data import Dataset, DataLoader
from PIL import Image
root = r"C:\Users\hq\Desktop\HoldingObject"
# 自定义图片图片读取方式,可以自行增加resize、数据增强等操作
def MyLoader(path):
return Image.open(path).convert