
pytorch
小星星*
举一反三 见微知著
展开
-
简单的一维线性回归-pytorch
代码来自书籍《深度学习入门之pytorch》,书里有几个错误,修改好了import matplotlib.pyplot as pltimport torchfrom torch import nnfrom torch import optimfrom torch.autograd import Variableimport numpy as npx_train = np.a...原创 2018-12-24 09:16:09 · 1046 阅读 · 0 评论 -
CAM方法获取显著图:基于pytorch的实现
CAM方法获取显著图:基于pytorch的实现1、下载代码2、具体代码(1)导入需要的模块(2)具体代码详解(3)生成结果转:CAM实践:基于pytorch的使用方法原文地址:https://cloud.tencent.com/developer/article/1370937非原创,直接copy的网上的,懒得打字,不喜勿喷1、下载代码git clone https://github.c...转载 2019-02-24 13:43:27 · 15915 阅读 · 6 评论 -
让网络能够训练任意大小的图片
为了让网络能够训练任意大小的图片,可以采用以下方法:Instead of sppnet, you can use this block of code in Pytorch to train a neural network with variable-sized inputs:#With these lines of code below, we can memorize the...原创 2019-06-25 14:46:19 · 1891 阅读 · 0 评论 -
pytorch 张量tensor,图片,CPU,GPU,数组等的转换
1, 创建pytorch 的Tensor张量:torch.rand((3,224,224)) #创建随机值的三维张量,大小为(3,224,224)torch.Tensor([3,2]) #创建张量,[3,2]2, cpu上的tensor和GPU即pytorch创建的tensor的相互转化b = a.cpu() # GPU → CPUa = b.cuda() #CPU → G...原创 2019-06-25 14:49:55 · 6570 阅读 · 0 评论