pytorch学习笔记
文章平均质量分 83
代码学习
不动脑筋
不动脑筋
展开
-
图像预处理相关代码学习
数据增强def augment(*args, hflip=True, rot=True): hflip = hflip and random.random() < 0.5 vflip = rot and random.random() < 0.5 rot90 = rot and random.random() < 0.5#三种操作执行概率均为50% def _augment(img): if hflip: img = img[:, ::原创 2021-03-19 21:03:31 · 965 阅读 · 0 评论 -
模型测试代码学习笔记
最近在看代码原创 2020-11-29 14:45:46 · 345 阅读 · 0 评论 -
利用nn.Module实现自定义层
class Flatten(nn.Module): def __init__(self): super(Flatten,self).__init__() def forward(self,input): return input.view(input.size(0),-1)Flatten()展平操作,进入全连接层之前使用类才能写进nn.Sequentialnn.ReLU----classF.relu----functionclas原创 2020-10-29 16:52:37 · 263 阅读 · 0 评论 -
pytorch---parameters学习
import torch.nn as nnnet = nn.Sequential(nn.Linear(4,2),nn.Linear(2,2))list(net.parameters())输出如下list(net.parameters())[0].shapetorch.Size([2, 4])list(net.parameters())[3].shapetorch.Size([2])存在转置的情况list(net.named_parameters())dict(net.name原创 2020-10-29 16:42:46 · 337 阅读 · 0 评论 -
pytorch视频笔记1-卷积神经网络基础
图片的表示输入一张灰度图用一个矩阵表示,矩阵的每个元素代表像素值,大小为0~255(常归一化为0~1)输入一张彩色图,有RGB三个通道,每张表存储相应的分量常用的表示方式全连接层的缺点以2828的手写字符为例,输入为7841的向量,中间节点256,输出为10个节点(一共10类)参数量784256 +256256+256256+25610=390K感受野借助于生物学的知识,人眼的局部视野 局部相关性例如图片中的小孩,在一个场景中首先注意到的是蛋糕,他不是一次性把场景中的所原创 2020-05-23 22:18:32 · 520 阅读 · 0 评论