栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

LeNet5 pytorch实现

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

LeNet5 pytorch实现

LeNet5

网络采用LeNet5结构,数据集为MNIST。输入为32x32的单通道图像。网络共分为7层。 网络结构 C1层-卷积层

核大小:5x5
核数目:6
步长:1
输出特征图大小:6x28x28
激活函数采用sigmoid

S2层-池化层

采用平均池化
核大小:2x2
步长:2
输出特征图大小:6x14x14

C3层-卷积层

核大小:5x5
核数目:16
步长:1
输出特征图大小:16x10x10
激活函数采用sigmoid

S4层-池化层

采用平均池化
核大小:2x2
步长:2
输出特征图大小:16x5x5

F5-全连接层

输入:400维
输出:120维

F6-全连接层

输入:120维
输出:84维

F7-全连接层

输入:84维
输出:10维


代码部分

LeNet5.py定义了网络结构
train.py训练模型,生成onnx文件。可以通过Neutron查看网络结构。

代码见github

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/355397.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号