
CNN模型总结
小小飞在路上
整理一系列python机器学习、神经网络代码(主要是基于TensorFlow,pytorch)等,有时候也会分享一些自己遇到的问题及解决方法。共同学习,共同进步,更新看情况(没办法,博主实在是太懒了)!!!
展开
-
pytorch加载VGG16及进行fine-tuning训练
加载VGG16模型并打印查看from torchvision import modelsnet=models.vgg16()print(net)1.1结果说明1.2查看某一部分**加载模型进行预训练,改变classifier层,固定feature层参数**2.1模型搭建import torchimport torch.nn as nnfrom torchvision import modelsfrom torchsummary import summarynet=m.原创 2020-11-18 19:34:51 · 6756 阅读 · 14 评论 -
网络结构总结1-Lenet:Lenet-5网络结构说明及pytorch实现
1.网络结构1.1论文结构1.2简化说明说明:共7层(3Conv+2pool+2FC)2.作用2.1各个部分的作用卷积:特征提取。(说明:同一卷积层中不同通道学习的是同一尺度下的多种特征,浅层卷积和深层卷积学习到的是多种尺度的特征,且深层次的卷积学习到的特征更抽象。)激活函数:引入非线性,提高模型的表达能力池化(降采样):特征压缩:降低模型复杂度防止过拟合+提取主要特征。不变性:translation(平移:不同位置),rotation(旋转:不同角度),scale(尺度:放缩)。原创 2020-11-04 17:01:02 · 1184 阅读 · 2 评论