pytorch的ImageNet类

官方文档：IMAGENET
源码：SOURCE CODE FOR TORCHVISION.DATASETS.IMAGENET

这个类可以帮助我们更方便使用ImageNet数据集，只需要下载好，然后提供数据集的根目录即可。
根目录下面应该包含这些文件：

ILSVRC2012_img_train.tar或者train文件夹
ILSVRC2012_img_val.tar或者val文件夹
ILSVRC2012_devkit_t12.tar.gz或者meta.bin

没有的话会报错。

源码解读

继承了ImageFolder类，ImageFolder继承了DatasetFolder，用来处理同种类的图片放在同一个文件夹的数据集。有一个class_to_idx字典和classes列表，classes就是root下面所有文件夹的名字，class_to_idx就是从class到label的映射，每个class对应的label就是将所有class按照字母顺序排序之后该class所处的位置或者说index（从0开始）。

pytorch的ImageNet类相关推荐

《pytorch车型细分类网络》的源码
说明:<pytorch车型细分类网络>.这篇文章代码有错误.我稍微调整了一下,可以正常跑了. 标题:pytorch动手实践:pytorch车型细分类网络 1)讲解,代码,主要参考知乎文章& ...
Pytorch实现102类鲜花分类(102 Category Flower Dataset)
Pytorch实现102类鲜花分类(VGG19和ResNet152模型) 本文主要讲解该算法的实现过程,原理部分需读者自行研究,可以找一些论文之类的. 实验环境 python3.6+pytorch1. ...
使用 Pytorch 进行多类图像分类
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达关于数据集此数据包含大小为150x150.分布在6个类别下的约2 ...
【深度学习】使用 Pytorch 进行多类图像分类
关于数据集此数据包含大小为150x150.分布在6个类别下的约25k图像. {'建筑物':0,'森林':1,'冰川':2,'山':3,'海':4,'街道':5} 训练.测试和预测数据在每个 zip ...
Pytorch:ToTensor(object)类
PyTorch在做一般的深度学习图像处理任务时,先使用dataset类和dataloader类读入图片,在读入的时候需要做transform变换,其中transform一般都需要ToTensor()操 ...
Pytorch —— nn.Module类(nn.sequential)
对于前面的线性回归模型. Logistic回归模型和神经网络,我们在构建的时候定义了需要的参数.这对于比较小的模型是可行的,但是对于大的模型,比如100 层的神经网络,这个时候再去手动定义参数就显得非 ...
pytorch torch.device类（表示在其上或将要分配torch.Tensor的设备）
from https://pytorch.org/docs/1.1.0/tensor_attributes.html?highlight=torch%20device#torch.torch.devi ...
Pytorch中DataLoader类
1 基本流程首先熟知,pytorch 的数据加载到模型的操作顺序是这样的: 创建一个 Dataset 对象创建一个 DataLoader 对象循环这个 DataLoader 对象,将img, l ...
ImageNet1K的下载与使用
0.前言 2.val集 2.1 下载 2.2 处理 0.前言 ImageNet不用多说,它包含了非常多的图片,总共有2w多个分类,但是显然太多.所以一般更常用的是ImageNet1K数据集,该数据集包 ...

pytorch的ImageNet类

源码解读

pytorch的ImageNet类相关推荐

最新文章

热门文章