本篇自学笔记来自于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