上上上周总结------重写Dataloader,自己的mydata代码

问题描述:

有的网上下载下来的数据集,数据和标签是混在一起的,不像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__
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值