pytorch自建数据集并显示(详细示例)
本文建dataset类方法参考:https://blog.csdn.net/u011995719/article/details/85102770
#通过自己定义.txt来形成list
1 代码
加载数据集 path = ‘E:/pycharmproject/dataset_read/dataset_method_2/dataset_test.txt’ 按照自己具体情况修改
import torch
import torchvision #用于下载数据集,进行图像增广操作等
import matplotlib.pyplot as plt #用于显示图片
from PIL import Image #用于读取数据
import numpy as np
import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
#忽略警告
import warnings
warnings.filterwarnings('ignore')
#选择运行设备
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
#定义自己的数据集类
class CustomDataset(torch.utils.data.Dataset):
def __init__(self, txt_path, transform=None, target_transform=None):
# TODO
# 1. Initialize file paths or a list of file names.
fh = open(txt_path, 'r')
imgs