深度学习
cvks
这个作者很懒,什么都没留下…
展开
-
pytorch学习笔记-----卷积,池化参数计算
卷积神经网络构建一般卷积层,relu层,池化层写成一个模块import torch.nn as nnclass CNN(nn.Module): def __init__(self): super(CNN,self).__init__() self.conv1=nn.Sequential( nn.Conv2d( in_channels=1,#输入通道数1(灰度图) ou原创 2021-07-11 21:29:56 · 234 阅读 · 0 评论 -
pytorch学习笔记------batchsize制作(分类任务)
基于pytorch框架训练batchsize制作from torchvision import datasetsimport osimport torchbatch_size=8#训练batchsizedata_dir=''#数据集文件夹image_datasets={x:datasets.ImageFolder(os.path.join(data_dir,x),data_transforms[x]) for x in ['train','valid']}#datasetdataloade原创 2021-07-11 20:56:07 · 634 阅读 · 0 评论 -
pytorch学习笔记-----数据增强
做法:对图像进行任意或特定角度的旋转,剪切等。意义:1.解决数据量不够问题2.高效利用数据from torchvision import transformsdata_transforms={ 'train': transforms.Compose([transforms.RandomRotation(45),#随机翻转 transforms.CenterCrop(224),#中心裁剪原创 2021-07-11 20:32:04 · 135 阅读 · 0 评论