
Pytorch
梦醒时分1218
这个作者很懒,什么都没留下…
展开
-
【torch】函数用法
\quad经常会用到torch,记录一下torch中一些函数的用法1. torch.max和F.softmaxpytorch中torch.max和F.softmax函数的维度解释原创 2020-03-29 22:23:57 · 264 阅读 · 0 评论 -
【Pytorch】--CIFAR10等数据集本地读取
1.问题\quad在使用Pytorch的时候,有时候需要在线下载数据集,因为在下载的过程中,封装好的代码,还要进行其他的操作(例如数据类型转换numpy->tensor),但是有时候因为下载网站在国外,进度条一直显示0%,\quad就像这样:2.解决办法step1.下载数据集到本地step2. 将本地存放CIFAR数据集路径放在浏览器下,回车step3. 修改c...原创 2020-02-10 17:42:10 · 6812 阅读 · 10 评论 -
【Pytorch】--torchvision.transforms.ToTensor
\quad 在使用Pytorch时候,需要导入图片,一般图片的类型都是numpy.ndarray,需要转换成tensor类型,这时候就要使用torchvision.transforms.ToTensor1.原理\quad 参考:Pytorch之深入torchvision.transforms.ToTensor与ToPILImage2.测试&&结果import cv2 a...原创 2020-02-10 00:08:07 · 1147 阅读 · 0 评论 -
【Pytorch】--MNIST手写体识别
1. 基于pytorch实现的MNIST手写体识别import torchimport torch.nn as nnfrom torch.autograd import Variableimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as plt# device = torch.de...原创 2020-02-09 02:49:30 · 528 阅读 · 0 评论