pytorch 之 imagefloder的用法

问题的关键在于如何确定图片和类别的对应关系

from torchvision.datasets import ImageFolderdataset = ImageFolder('F:/my_code/data/0dB/train/')

读入数据

print(dataset.imgs[2])

此时，dataset.imgs是一个列表，里面有很多元素，选中第2个
输出的是

('F:/my_code/data/0dB/train/BFSK\\BFSK100.jpg', 0)

也就是说，imagefolder应该是以文件夹名字的顺序进行排列的
BFSK对应的0
BPSK对应的1
等等……

pytorch 之 imagefloder的用法相关推荐

Pytorch 中retain_graph的用法
Pytorch 中retain_graph的用法用法分析在查看SRGAN源码时有如下损失函数,其中设置了retain_graph=True,其作用是什么? #################### ...
【Pytorch】 repeat()的用法详解
Pytorch的repeat()方法再深度学习中经常用到,用于复制tensor,最好的说明当然是官方文档. repeat的用法说明很简单:重复每个张量的维度的次数. -这里有个warrning很有意思 ...
pytorch笔记——torch.randperm用法
前言记录randperm用法. 方法介绍 torch.randperm(n) 这个方法将[0, n)中的元素随机排列,函数名randperm是random permutation缩写. permut ...
pytorch torch.sort()的用法以及例子
1.作用对tensor中元素排序 2.用法 dim = -1,按照行排序,dim= 1按照列排序,descending=True,则递减排序,否则递增 3.例子按照行排序 logits = tor ...
PyTorch torch.Tensor.contiguous() 用法与理解
中文文档: contiguous() → Tensor 返回一个内存连续的有相同数据的 tensor,如果原 tensor 内存连续则返回原 tensor 英文文档: contiguo ...
pytorch中contiguous()的用法
contiguous:view只能用在contiguous的variable上.如果在view之前用了transpose, permute等,需要用contiguous()来返回一个contiguou ...
Pytorch之gather的用法（易懂的方式解释）
百度了一圈gather的用法,看的一知半解,结合了几篇博客的讲解,终于理解了这个的用法,记录下来,用于以后忘记的时候自己可以快速复习,同时不懂得小伙伴也可以参考下我这得理解,或许能帮助到你!!! 首先 ...
pytorch中arange()函数用法
语法:torch.arange(start=0,end,step=1) 解释:开始默认为0,步长默认为1,可以不写:终值必须写返回的个数: 举例:
Pytorch permute()的简单用法
作用:调换Tensor中各维度的顺序输入:数组,用来表示调换后的各维度顺序.一开始没修改的顺序其实就是[0,1,2-] 例子:有一个6×7×8×9的Tensor,我们想将其第2维(8)和第三维(9) ...

pytorch 之 imagefloder的用法

pytorch 之 imagefloder的用法相关推荐

最新文章

热门文章