Dataset和DataLoader的书写

---

书写Dataset和DataLoader

说明：搭建网络需要数据的读入和预处理，model的构建，损失函数和优化策略的选择。该博客以PIL读取图像，构建最基本的Dataset和DataLoader。

---

一、导入必要的库

import torch
from torch.utils.data import Dataset
from torch.utils.data import DataLoader
import torchvision.transforms as tr

import os 
from PIL import Image 

用PIL读取图像为例，还可以用matplotlib和cv2读取图像

二、继承Dataset自定义数据读写

class MyDataset(Dataset):
    def __init__(self,root_path):
        filenames = sorted(os.listdir(root_path))#filenames是列表 相当于存储的路径下的所有子文件名（查阅os.listdir）
        self.transforms = tr.Compose([
            tr.CenterCrop(256),#剪裁成等大的图像块 因为之后batch操作时必须保证每个batch中图像大小一致（尝试注释掉这行，看报错信息）
            tr