【学习】torchvision.datasets.ImageFolder()

在分类任务中,数据集文件存储往往是如下形式:

- train- class1- image1.jpg- image2.jpg...- class2- image1.jpg- image2.jpg......

此时,我们想要获取图片和标签,标签即为文件名(class1、class2…)
可以使用torchvision.datasets.ImageFolder()来进行获取,示例代码如下:

dataset = datasets.ImageFolder(root=DATA_PATH/'train')

torchvision.datasets.ImageFolder() 参数列表:

  • root:图像文件读取路径
  • transform:对图像数据采取的数据增强策略
  • target_transform:对label进行转换
  • loader:指定加载图像的函数
  • is_valid_file:获取图像路径,检查文件的有效性

返回值

dataset 返回有如下三个属性:

  • self.classes:用一个 list 保存类别名称
  • self.class_to_idx:类别对应的索引,与不做任何转换返回的 target 对应
  • self.imgs:保存(img-path, class) tuple的 list

我们得到的dataset,它的结构就是[(img_data,class_id),(img_data,class_id),…]下面我们打印dataset第一个元素中的图片:
在这里插入图片描述
返回对应的label:
在这里插入图片描述
其中,对于dataset[0]来说,其中也存储了两个元素,第一个是图片,第二个是类别索引号。

sample = dataset[0]
img = sample[0]   #图片
label = sample[1]  #类别索引

注意:

  1. dataset中存储的label是按文件夹顺序生成对应索引的,且以下标为0开始。如果要读取类别的字符,可以通过self.classes[0]来获取。
  2. train文件夹下的文件格式是固定的,不能有多余的文件,否则会读取出错。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/724709.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【DevSecOps】2024 年需要警惕的 10 大 Web 应用程序安全威胁

【DevSecOps】2024 年需要警惕的 10 大 Web 应用程序安全威胁 由于 2023 年出现了许多创新,我们之前所了解的许多内容都发生了巨大变化;随着其中一些重大变化,威胁格局也发生了转变,一些旧威胁减少了,一些新威胁增加了。 技术每天都在不断变化,当我们谈论技术和相关威胁…

24计算机考研调剂 | 山东科技大学

山东科技大学王瑞副教授团队(先进成形技术与装备)硕士招生 考研调剂招生信息 学校:- 专业:工学->机械工程->机械制造及其自动化 工学->材料科学与工程->材料加工工程 工学->治金工程->钢铁冶金 工学->计算机科学与技术 年级:20…

AI大模型:创新前沿的探索之路

AI大模型一直被视为人工智能领域的创新前沿,它们拥有强大的计算能力和学习能力,能够在各种复杂的任务中表现出色。随着技术的不断进步,越来越多的研究者和企业开始投入到AI大模型的研发和应用中,希望能够探索出更多的可能性。 在…

Tomcat基础与Nginx的动静分离

一、TOMCAT基础功能 (一)自动解压war包 在配置文件中讲到,当接受到请求后,会匹配符合要求的Host,在配置文件中的Host只有一个,且规定了自动解压war包 自动解压war包 .war:WebApp打包,类zip格…

stl的基本知识学习

1.vector: 2.set: 3.map: 4.栈: 5.队列: 6. unordered_map与unordered_set: 7. 位运算: 8.cctype: 导图:

ARM中专用指令(异常向量表、异常源、异常返回等)

状态寄存器传送指令 CPSR寄存器 状态寄存器传送指令:访问(读写)CPSR寄存器 读CPSR MRS R1, CPSR R1 CPSR 写CPSR MSR CPSR, #0x10 0x10为User模式,且开启IRQ和FRQ CPSR 0x10 在USER模式下不能随意修改CPSR,因为USER模式…

VMvare17安装centos8安装宝塔面板 教程

阿里镜像站:https://mirrors.aliyun.com/centos centos-8-isos-x86_64安装包下载_开源镜像站-阿里云 https://mirrors.aliyun.com/centos/8/isos/x86_64/CentOS-8.5.2111-x86_64-dvd1.iso 将上面的链接复制到迅雷进行高速下载 vmvare安装配置教程安装教程 CentOS…

动态规划(算法竞赛、蓝桥杯)--线性DP股票买卖含冷冻期

1、B站视频链接&#xff1a;E24 线性DP 股票买卖含冷冻期_哔哩哔哩_bilibili #include <bits/stdc.h> using namespace std; const int N100010; int w[N],f[N][3];int main(){int n;cin>>n;for(int i1;i<n;i)cin>>w[i];f[0][1]-1e7;f[0][0]-1e7;f[0][2]…

svg背景适应元素大小

svg背景适应元素大小 background-image: url("./test.svg"); background-size: 100% 100%;如果你的css代码是以上这样的&#xff0c;它可能不会达到理想的效果 为了解决这个问题 方法一&#xff1a;修改svg文件&#xff0c;在svg标签中加属性preserveAspectRatio“…

Django学习记录08——图表及文件上传案例

1.图表Echarts的应用 Apache ECharts 1.1 使用方法 引用echarts.js即可到官方文档中查询使用 1.2 常用图标的使用 图表展示页面的部署&#xff08;主要展示折线图、柱状图、饼图&#xff09; {% block content %}<div class"container"><div class&qu…

Docker安装MySQL镜像实战分享

今天我们对Docker安装MySQL镜像进行实战分享&#xff0c;以更深入的了解容器的使用场景。我们在云付服务器Ubuntu环境上已经安装好了Docker&#xff0c;接下来我们开始安装mysql5.7版本&#xff0c;安装mysql有两种思路&#xff0c;直接拉取mysql镜像和自己做mysql镜像&#xf…

JAVA语言编写一个方法,两个Long参数传入,使用BigDecimal类,计算相除四舍五入保留2位小数返回百分数。

在Java中&#xff0c;你可以使用BigDecimal类来执行精确的浮点数计算&#xff0c;并且可以指定结果的小数位数。以下是一个方法&#xff0c;它接受两个Long类型的参数&#xff0c;并使用BigDecimal来计算它们的商&#xff0c;然后将结果四舍五入到两位小数&#xff0c;并返回一…

对猫毛过敏还想养猫怎么办?除毛好的宠物空气净化器品牌推荐

许多朋友喜欢猫咪&#xff0c;但与猫咪相处一段时间后&#xff0c;他们可能会出现鼻塞、打喷嚏和眼泪不断的情况。让我们来科普一下如何让那些容易过敏的家人与猫咪更好地相处吧。为什么会过敏呢&#xff1f;因为猫咪的唾液中含有一种叫做Fel d1的蛋白质&#xff0c;通过舔毛散…

GEE 依照范围裁剪 下载Sentinel-2数据

0. GEE介绍 Google Earth Engine&#xff08;GEE&#xff09; 是由Google开发的一种云端平台&#xff0c;旨在提供强大的地理空间数据处理和分析工具。GEE集成了大量的遥感影像数据和地理空间数据集&#xff0c;以及高性能的计算资源&#xff0c;使用户能够在云端高效地进行大规…

AI的归纳和演绎法

AI的归纳和演绎法分别是什么&#xff1f; AI的归纳和演绎法是两种常见的推理方法。 归纳法&#xff08;inductive reasoning&#xff09;是一种从特殊到一般的过程&#xff0c;在有限的实例观察中得出一般规律或原则。用简单的说法&#xff0c;就是从一些具体的事物或情况中总…

GDB调试入门笔记

文章目录 What&#xff1f;WhyHow安装GDB安装命令查看是否安装成功调试简单的程序预备一个程序调试 使用breakinfolistnextprintstep一些小技巧在gdb前shell日志功能watch point| catch point 调试core调试一个运行的程序 What&#xff1f; GDB是什么&#xff1f; 全称GNU sym…

1、MQ_介绍、优缺点、类型等

MQ介绍 1. MQ概述 MQ&#xff08;Message Queue&#xff09;&#xff1a;消息队列&#xff0c;是基础数据结构中FIFO&#xff08;first in first out&#xff09;的一种数据结构。一般用来解决流量削峰、应用解耦、异步处理等问题&#xff0c;实现高性能&#xff0c;高可用&a…

24计算机考研调剂 | 中国民用航空飞行学院

中国民用航空飞行学院计算机学院2024年研究生招生啦~ 考研调剂招生信息 学校:中国民航飞行学院 专业:工学->电子信息 年级:2024 招生人数:- 招生状态:正在招生中 联系方式:********* (为保护个人隐私,联系方式仅限APP查看) 补充内容 中国民用航空飞行学院计算机学院…

Kafka|处理 Kafka 消息丢失的有效措施

文章目录 消息丢失场景生产者端Kafka Broker消费者端 如何防止消息丢失生产者端Kafka Broker 端消费者端 扩展如何实现消费端的重试功能&#xff1f;有如何处理消息重复&#xff1f; 消息丢失是 Kafka 系统中一个严重的问题&#xff0c;可能会发生在生产者、Broker 或消费者任何…

鸿蒙实战开发:数据交互【RPC连接】

概述 本示例展示了同一设备中前后台的数据交互&#xff0c;用户前台选择相应的商品与数目&#xff0c;后台计算出结果&#xff0c;回传给前台展示。 样例展示 基础信息 RPC连接 介绍 本示例使用[ohos.rpc]相关接口&#xff0c;实现了一个前台选择商品和数目&#xff0c;后台…