Pytorch数据读入之Dataset

Pytorch数据读入之Dataset

Dataset:提供一种方式,获取数据和label

from torch.utils.data import Dataset
from PIL import Image  ## 读取图片的库,可以对图片进行可视化
import os  ## 关于系统操作的库,主要用来对文件路径操作

# dir_path='dataset/hymenoptera_data/train/ants'   # 获取数据相对路径/也可以是绝对路径
# import os          ## 对数据所在文件路径进行字符串操作
# img_path_list=os.listdir(dir_path)  ## 将地址传给os.listdir会吧所有数据地址汇成一个列表,我的数据地址是dir_path='dataset/hymenoptera_data/train/ants' 

# img_path_list[0]   #  这个列表存的就是每个数据的文件名,我们取出第一个看一下
# 输出:'0013035.jpg'  # 和列表操作一样可以逐个读取



创建读取数据的class类

class MyData(Dataset):
    '''
    读数据的类需要继承Dataset这个类
    '''

    def __init__(self, root_dir, label_dir):  # 初始化主要是为这个所建的类提供全局变量
        
        '''
        :param root_dir: 数据集根目录
        :param label_dir: 数据集标签的目录,这个只填标签目录即可,我们有函数可以合并这两个目录

        ####################################
        root_dir='dataset/hymenoptera_data/train'
        label_dir='ants'
        path=os.path.join(root_dir,label_dir)  os.path.join()这个函数可以拼接两个文件路径形成一个文件路径
        输出:'dataset/hymenoptera_data/train\\ants'       我的数据就是放在dataset/hymenoptera_data/train\\ants里面\\环境里需要用\\
        '''
        
        self.root_dir = root_dir  ## self 的作用就是把定义的变量设置成全局变量
        self.label_dir = label_dir
        self.path = os.path.join(self.root_dir, self.label_dir)  ## 获取数据地址,只不过这个数据文件名称是标签
        self.img_path = os.listdir(self.path)

    def __getitem__(self, idx):  # item原本是item,在这里我们使用idx
        
        # 取图片中的某一个
        '''
        img_name = img_path[idx]
        print(img_name)
        0013035.jpg
        '''

        img_name = self.img_path[idx]  ## 根据idx索引来提取索引所☞的图片的编号
        img_item_path = os.path.join(self.root_dir, self.label_dir,
                                     img_name)  # 每一个图片的位置'dataset/hymenoptera_data/train\\ants\\0013035.jpg'
        ## 读取图片
        img = Image.open(img_item_path)
        ## 标签
        label = self.label_dir

        return img, label

    def __len__(self):
        '''
        返回数据集长度,数据集的长度就是列表文件的长度
        :return:
        '''

        return len(self.img_path)

实例化演示

这里获取,蚂蚁的数据集

root_dir = 'dataset/hymenoptera_data/train'
ant_label_dir = 'ants'
ant_dataset = MyData(root_dir, ant_label_dir)

这里获取,蜜蜂的数据集

bees_dir = 'dataset/hymenoptera_data/train'
bees_label_dir = 'bees'
bees_dataset = MyData(bees_dir, bees_label_dir)

整个数据集应该是,ants和bees的集合而不是单独的所以我们要进行整合

train_dataset = ant_dataset + bees_dataset

热门文章

暂无图片
编程学习 ·

exe4j详细使用教程(附下载安装链接)

一、exe4j介绍 ​ exe4j是一个帮助你集成Java应用程序到Windows操作环境的java可执行文件生成工具,无论这些应用是用于服务器,还是图形用户界面(GUI)或命令行的应用程序。如果你想在任务管理器中及Windows XP分组的用户友好任务栏…
暂无图片
编程学习 ·

AUTOSAR从入门到精通100讲(126)-浅谈车载充电系统通信方案

01 引言 本文深入研究车载充电系统策略,设计出一套基于电动汽车电池管理系统与车载充电机的CAN通信协议,可供电动汽车设计人员参考借鉴。 02 电动汽车充电系统通讯网络 电动汽车整车控制系统中采用的是CAN总线通信方式,由一个整车内部高速CAN网络、内部低速CAN网络和一个充电…
暂无图片
编程学习 ·

CMake(九):生成器表达式

当运行CMake时,开发人员倾向于认为它是一个简单的步骤,需要读取项目的CMakeLists.txt文件,并生成相关的特定于生成器的项目文件集(例如Visual Studio解决方案和项目文件,Xcode项目,Unix Makefiles或Ninja输入文件)。然…
暂无图片
编程学习 ·

47.第十章 网络协议和管理配置 -- 网络配置(八)

4.3.3 route 命令 路由表管理命令 路由表主要构成: Destination: 目标网络ID,表示可以到达的目标网络ID,0.0.0.0/0 表示所有未知网络,又称为默认路由,优先级最低Genmask:目标网络对应的netmaskIface: 到达对应网络,应该从当前主机哪个网卡发送出来Gateway: 到达非直连的网络,…
暂无图片
编程学习 ·

元宇宙技术基础

请看图: 1、通过AR、VR等交互技术提升游戏的沉浸感 回顾游戏的发展历程,沉浸感的提升一直是技术突破的主要方向。从《愤怒的小鸟》到CSGO,游戏建模方式从2D到3D的提升使游戏中的物体呈现立体感。玩家在游戏中可以只有切换视角,进而提升沉浸…
暂无图片
编程学习 ·

flink的伪分布式搭建

一 flink的伪分布式搭建 1.1 执行架构图 1.Flink程序需要提交给 Job Client2.Job Client将作业提交给 Job Manager3.Job Manager负责协调资源分配和作业执行。 资源分配完成后,任务将提交给相应的 Task Manage。4.Task Manager启动一个线程以开始执行。Task Manage…
暂无图片
编程学习 ·

十进制正整数与二进制字符串的转换(C++)

Function one: //十进制数字转成二进制字符串 string Binary(int x) {string s "";while(x){if(x % 2 0) s 0 s;else s 1 s;x / 2;}return s; } Function two: //二进制字符串变为十进制数字 int Decimal(string s) {int num 0, …
暂无图片
编程学习 ·

[含lw+源码等]微信小程序校园辩论管理平台+后台管理系统[包运行成功]Java毕业设计计算机毕设

项目功能简介: 《微信小程序校园辩论管理平台后台管理系统》该项目含有源码、论文等资料、配套开发软件、软件安装教程、项目发布教程等 本系统包含微信小程序做的辩论管理前台和Java做的后台管理系统: 微信小程序——辩论管理前台涉及技术:WXML 和 WXS…
暂无图片
编程学习 ·

树莓派驱动DHT11温湿度传感器

1,直接使用python库 代码如下 import RPi.GPIO as GPIO import dht11 import time import datetimeGPIO.setwarnings(True) GPIO.setmode(GPIO.BCM)instance dht11.DHT11(pin14)try:while True:result instance.read()if result.is_valid():print(ok)print(&quo…
暂无图片
编程学习 ·

ELK简介

ELK简介 ELK是三个开源软件的缩写,Elasticsearch、Logstash、Kibana。它们都是开源软件。不过现在还新增了一个 Beats,它是一个轻量级的日志收集处理工具(Agent),Beats 占用资源少,适合于在各个服务器上搜集日志后传输给 Logstas…
暂无图片
编程学习 ·

Linux 基础

通常大数据框架都部署在 Linux 服务器上,所以需要具备一定的 Linux 知识。Linux 书籍当中比较著名的是 《鸟哥私房菜》系列,这个系列很全面也很经典。但如果你希望能够快速地入门,这里推荐《Linux 就该这么学》,其网站上有免费的电…
暂无图片
编程学习 ·

Windows2022 无线网卡装不上驱动

想来 Windows2022 和 windows10/11 的驱动应该差不多通用的,但是死活装不上呢? 搜一下,有人提到 “默认安装时‘无线LAN服务’是关闭的,如果需要开启,只需要在“添加角色和功能”中,选择开启“无线LAN服务…
暂无图片
编程学习 ·

【嵌入式面试宝典】版本控制工具Git常用命令总结

目录 创建仓库 查看信息 版本回退 版本检出 远程库 Git 创建仓库 git initgit add <file> 可反复多次使用&#xff0c;添加多个文件git commit -m <message> 查看信息 git status 仓库当前的状态git diff 差异对比git log 历史记录&#xff0c;提交日志--pret…
暂无图片
编程学习 ·

用Postman生成测试报告

newman newman是一款基于nodejs开发的可以运行postman脚本的工具&#xff0c;使用Newman&#xff0c;可以直接从命令运行和测试postman集合。 安装nodejs 下载地址&#xff1a;https://nodejs.org/en/download/ 选择自己系统相对应的版本内容进行下载&#xff0c;然后傻瓜式安…
暂无图片
编程学习 ·

Java面向对象之多态、向上转型和向下转型

文章目录前言一、多态二、引用类型之间的转换Ⅰ.向上转型Ⅱ.向下转型总结前言 今天继续Java面向对象的学习&#xff0c;学习面向对象的第三大特征&#xff1a;多态&#xff0c;了解多态的意义&#xff0c;以及两种引用类型之间的转换&#xff1a;向上转型、向下转型。  希望能…