ImageFolder使用方法

ImageFolder假设所有的文件按文件夹保存好，每个文件夹下面存贮同一类别的图片，文件夹的名字为分类的名字
例如：
载入图片：

train_dataset=ImageFolder('./time_fequence_image/')

此时，看看这个train_dataset的类型：

print(type(train_dataset))

输出结果为：

<class 'torchvision.datasets.folder.ImageFolder'>

查看train_data里面的类别：

print(train_dataset.class_to_idx)

输出就是文件夹的名字和序号：

{'BPSK': 0, 'CW': 1, 'LMF': 2, 'NLMF': 3}

这是一个字典结构的。

print(train_dataset[0][0])

#所有图片的路径和对应的label
print(dataset.imgs)

#没有任何转变，所有返回的还是PIL Image对象
print(dataset[0][1]) #第二维度为1 ，表示label
print(dataset[0][0]) #第二维度为0，表示图片数据

print(dataset.classes)  #根据分的文件夹的名字来确定的类别，就是文件夹的名字，列表结构
print(dataset.class_to_idx) #按顺序为这些类别定义索引为0,1...，字典结构
print(dataset.imgs) #返回从所有文件夹中得到的图片的路径以及其类别，列表结构

ImageFolder使用方法相关推荐

spikingjelly的20201221版本跑通ANN2SNN
使用自己的数据集最大的一个问题在于,源代码使用的是MNIST数据集,而我使用的是自己构建的图片集. 主要区别在以下几点: 1. 源代码数据加载使用的方法为: # train_data_dataset ...
PyTorch 07—计算机视觉基础
卷积神经网络(CNN)简介前面我们讲解了机器学习基础知识,包括多层感知器等问题.下面我们要介绍的目标识别与分类,就是在前面问题的基础上进行扩展,实现对于图像等分类和识别.实现对图像的高准确率识别离不 ...
Pytorch图像分类实战笔记Task01-Task07|Datawhale组队学习
学习计划 Task01:构建自己的图像分类数据集第一节配套的代码:(按照顺序学习) 1.1配置环境 tqdm库官方说明:https://github.com/tqdm/tqdm#documentat ...
低照度图片增强——传统算法
文章目录前言一.基于直方图的算法 1.1直方图均衡化的增强算法 1.2直方图规定化的增强算法二.基于图像变换的算法三. 基于Retinex理论的增强算法 3.1 单尺度Retinex算法 (S ...
在 Oracle Enterprise Linux 和 iSCSI 上构建您自己的 Oracle RAC 11g 集群
作者:Jeffrey Hunter 了解如何以低于 2,700 美元的费用在 Oracle Enterprise Linux 上安装并配置 Oracle RAC 11g 第 2 版开发集群. 本指南中 ...
Java面试题大全2021版
一.Java 基础 JDK 和 JRE 有什么区别? JDK:Java Development Kit 的简称,java 开发工具包,提供了 java 的开发环境和运行环境. JRE:Java Run ...
手把手教你使用深度学习的方法进行人脸解锁
来源:DeepHub IMBA 本文约3000字,建议阅读8分钟本文手把手教你如何创建人脸解锁算法. 今天,我们将使用深度学习来创建面部解锁算法.要完成我们的任务需要三个主要部分. 查找人脸的算法 ...
pytorch加载自己的图片数据集的两种方法
目录 ImageFolder 加载数据集使用pytorch提供的Dataset类创建自己的数据集. Dataset加载数据集接下来我们就可以构建我们的网络架构: 训练我们的网络: 保存网络模型(这 ...
Pytorch中torchvision.datasets.ImageFolder报错：Found 0 files in subfolders of:XXX，完美解决！
1.报错如下: RuntimeError: Found 0 files in subfolders of:XXX Supported extensions are: .jpg,.jpeg,.png,. ...

ImageFolder使用方法

ImageFolder使用方法相关推荐

最新文章

热门文章