pytorch学习笔记(一)---Dataset类

          本篇自学笔记来自于b站《PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】》,Up主讲的非常通俗易懂,文章下方有视频连接,如有需要可移步up主讲解视频,如有侵权,实非故意,深表歉意,请与我联系,删除相关内容! 

        再一次表示感谢!

  • Dataset类:提供了一种方式去获取数据及其label
  • 主要应用:
    • 如何获取每一个数据和对应的label
    • 如何获取数据的长度
  • 引入
from torch.utils.data import Dataset
  • 例子:
#展示Dataset类的用法:提供了一种获取数据的和对应label的方式
""""
    1.如何获取每一个数据及其label
    2.获取数据的长度
"""
from torch.utils.data import Dataset  #引入对应的package
import os
from PIL import Image #图片操作的package

class Dataread(Dataset):#建立一个类,要继承Dataset类
    def __init__(self,root_path,images_path,labels_path):
        self.root_path = root_path
        self.images_path = images_path
        self.labels_path = labels_path
        self.img_path
  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值