pytorch中几种Dataset读取数据重写实现
通过重写Dataset类,对自己制作的数据集进行读取后传给DataLoader。主要用来完成从哪里读取数据和标签的功能。主要是__getitem__(返回数据集和标签)和__len__(返回数据的长度)这两个方法。
import numpy as np
import torch
import os
from PIL import Image
from torch.utils.data import Dataset
class MyDataset_1(Dataset):
"""
通过包含数据
原创
2021-06-07 15:36:23 ·
1925 阅读 ·
0 评论