Pytorch深度学习者必备技能教程(五)
5页1、搭建网络思路:mnist数据集下载与处理搭建神经网络损失函数与优化函数的选择定义可视化函数(不是本文的重点)模型训练以及结果图像绘制模型测试结果可视化与结果分析代码实现1. 导入需要用到的安装包与模块:这里我么直接使用Pytorch自带的mnist数据集,所以要用到torchvision安装包。fromtorch.utilsimportdatafromtorchvisionimportdatasets,transformsfromtorch.nnimportSequential,Conv2d,ReLU,MaxPool2d,Linear,CrossEntropyLossfromtorch.optimimportAdamfromtorch.autogradimportVariablefrommatplotlibimportcm,pyplotaspltfromsklearn.manifoldimportTSNE#将最后一层的输出降维处理,方便结果的可视化importtorch.nnasnnimporttorchimportos2. mnist数据集下载与处理:直接下载下来的mnist数据集并
2、不符合输入给网络的格式,所以我们需要对数据集做一些处理,以满足模型训练的要求。由于小编电脑配置较低,测试数据量为1800张图片。LR=0.01BATCH_SIZE=32DOWNLOAD_MINIST=FalseEPOCHS=1HAS_SK=True#创建数据集目录ifnotos.listdir(./model_datas/MnistDatasets/):DOWNLOAD_MINIST=True#训练数据集处理train_data=datasets.MNIST(root=./model_datas/MnistDatasets/,#数据集需要保存的路径train=True,#true:训练数据False:测试数据transform=transforms.ToTensor(),#将PIL.Image或numpy.ndarray数据转换为形状为torch.FloatTensor(CxHxW),同时归一化download=DOWNLOAD_MINIST#下载数据集,如果有,就直接加载,如果如果没有,就去下载)#将输入图片的shape转化为(50,1,28,28)data_loader=data.
3、DataLoader(dataset=train_data,batch_size=BATCH_SIZE,shuffle=True,)test_data=datasets.MNIST(root=./model_datas/MnistDatasets/,train=False,)test_x=torch.unsqueeze(test_data.test_data:1800,dim=1).type(torch.FloatTensor)/255.#将测试数据的输入shape由原来的(10000,28,28)转化为(10000,1,28,28)并且将输出数据归一化test_y=test_data.test_labels:1800#提取测试数据的标签值3. 搭建神经网络:将模型数据预处理完之后,接下来才是我们这篇文章的重点,搭建一个卷积神经网络。输入层数据格式:32,1,28,28第一卷积层:卷积层:shape32,1,28,2832,16,28,28这里的卷积层只是将原来的通道数由1变为16,图片大小没有变化激活层:这里我们选择Relu作为激活函数对卷积层输出的数据进行去线性。池化层:shape
4、32,16,28,2832,16,14,14池化层不改变通道数,只改变图片的大小第二卷积层:卷积层:shape32,16,14,1432,32,14,14这里的卷积层只是将上一层池化层输出的通道数由16变为32,图片大小没有变化激活层:选择Relu作为激活函数对卷积层输出的数据进行去线性。池化层:shape32,32,14,1432,32,7,7池化层不改变图片通道数,只改变图片的大小全连接层:数据降维处理:32,32,7,732,32*7*7首先将第二卷积层池化层的输出降维处理为全连接层可以接收的数据格式全连接层:shape32,32*7*732,10将降维处理完的数据格式作为全连接层的输入,使用一个32*7*7,10的权重,将结果输出;具体的网络结果如下图所示:classCNN(nn.Module):def_init_(self):super(CNN,self)._init_()self.conv_1=Sequential(#第一卷积层Conv2d(in_channels=1,out_channels=16,kernel_size=5,stride=1,padding=2),ReL
《Pytorch深度学习者必备技能教程(五)》由会员鲁**分享,可在线阅读,更多相关《Pytorch深度学习者必备技能教程(五)》请在金锄头文库上搜索。
关于工商管理的论文
五年级译林版英语下册连词成句家庭专项练习
公司管理系统规章规章制度(员工守则+员工行为要求规范+员工管理系统规章制度)
企业新人试用期转正工作总结「二」
2022年医院信息系统应急演练方案
暑假周记
露天管道外壁配套方案及施工工艺
市局督查组对县国土资源工作情况的督查报告工作报告报告总结
企业周年庆典祝福语_企业职员生小孩祝福语
企划方案
锦纶弹力丝项目商业计划书写作参考
企业策划案例
引水隧洞衬砌工程施工组织设计方案
大班教师教育随笔200篇
【合伙办厂】合伙协议书格式、范文、范本、样本、规范
旅游公司企业口号大全
机电传动与控制课程综合训练项目报告模板2
春节初二作文
自信成就成功
理实佳讯四环营销咨询项目—营销诊断访谈提纲(高层)
2023-11-27 11页
2023-04-22 2页
2022-10-05 143页
2023-03-13 44页
2022-09-30 10页
2024-02-23 39页
2023-12-13 71页
2023-04-15 1页
2023-04-23 63页
2023-08-20 60页