首页 > 其他分享 >6、dataset实战

6、dataset实战

时间:2023-01-26 21:22:48浏览次数:42  
标签:实战 __ os label dataset root dir

1、首先从包中导入dataset类(from torch.utils.data import Dataset)

2、定义一个子类去继承dataset类,可以重写里面的方法

  (1)__init__(self):初始化方法(必写,因为创建一个实例的时候必须要进行初始化)

  (2)__getitem__(self,item):获取每一条数据的方法

    可以在pycharm下面的python console控制台里来边编写边查看

  

首先获取图片地址(通过索引来获取,必须定义个列表来存放这些地址)----------import os  引入os库,

 

 

 

def __init__(self,root_dir,label_dir):
# 为整个类提供全局变量,root_dir用来获取存放图片的文件夹地址,label_dir代表图片标签的文件夹地址

root_dir="learn_Pytorch/dataset/hymenoptera_data/train"
label_dir="learn_Pytorch/dataset/hymenoptera_data/train/ants"
path=os.path.join(root_dir,label_dir);    os.path.join()这个函数的功能就是将上面两个地址进行拼接

 

 

 

标签:实战,__,os,label,dataset,root,dir
From: https://www.cnblogs.com/ar-boke/p/17068221.html

相关文章