问题描述:
有的网上下载下来的数据集,数据和标签是混在一起的,不像torchvision里下载好的那样数据和标签给咱们做好分类,所以需要我们重写DataLoader这个函数。
例子:以图像分割数据集为例, 使用如下数据集重写我们的dataloader:
数据集来源:
Xiaoyong Shen, Xin Tao, Hongyun Gao, Chao Zhou, Jiaya Jia. Deep Automatic Portrait Matting. European Conference on Computer Vision (ECCV),2016.
图一 训练集中label和picture分别一个文件夹
图二 标签的文件名称 #####_matte
图三 图片
1.导入包:
import os
from torch.utils.data import DataLoader
from PIL import Image
from torch.utils.data.dataset import Dataset
import torchvision.transforms as transforms
2.Mydata 类
必须要有的函数 len 和 getitem
''' 数据集处理 '''
class Mydata(Dataset):
def __init__