博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
torchvision的安装及使用
阅读量:5148 次
发布时间:2019-06-13

本文共 738 字,大约阅读时间需要 2 分钟。

1. 根据pytorch的命令安装torchvision

2. pytorch的官网提供的例子中使用了cifar-10数据集,需要下载。

    通过torchvision下载会很慢,且充满不确定性,我试了诸多次最后才成功。

  一旦成功后,尽管download = true,也不会再下载了。

  Files already downloaded and verified

    Files already downloaded and verified

    还可以直接去官网下载,下载,然后解压为cifar-10-batches-py文件夹,

    并复制到相对目录./data下,这个./data为工程的目录,

    例如使用pycharm创建了一个hello的工程:/Users/Aey/PycharmProjects/hello/venv/include/。

3.  torchvision.datasets.CIFAR10的参数

  root,表示cifar10数据的加载的相对目录

  train,表示是否加载数据库的训练集,false的时候加载测试集

  download,表示是否自动下载cifar数据集

  transform,表示是否需要对数据进行预处理,none为不进行预处理

4. torch.utils.data.DataLoader的参数

    batch_size=4  将训练集的50000张图片划分成12500份,每份4张图,用于mini-batch输入。

    shffule=True在表示不同批次的数据遍历时,打乱顺序。

    num_workers=2表示使用两个子进程来加载数据

转载于:https://www.cnblogs.com/anenyang/p/8425910.html

你可能感兴趣的文章
LibSVM for Python 使用
查看>>
Centos 7.0 安装Mono 3.4 和 Jexus 5.6
查看>>
CSS属性值currentColor
查看>>
java可重入锁reentrantlock
查看>>
浅谈卷积神经网络及matlab实现
查看>>
解决ajax请求cors跨域问题
查看>>
《收获,不止Oracle》pdf
查看>>
Real-Time Rendering 笔记
查看>>
如何理解HTML结构的语义化
查看>>
Activity之间的跳转:
查看>>
实验四2
查看>>
多路复用
查看>>
Python数据可视化之Pygal(雷达图)
查看>>
Java学习笔记--字符串和文件IO
查看>>
转 Silverlight开发历程—(画刷与着色之线性渐变画刷)
查看>>
SQL语法(3)
查看>>
在js在添版本号
查看>>
sublime3
查看>>
Exception Type: IntegrityError 数据完整性错误
查看>>
Nuget:Newtonsoft.Json
查看>>