Deep Learning---caffe模型参数量(weights)计算

本文介绍了如何使用draw_convnet工具绘制深度学习网络计算图,重点讨论了特征图大小计算,特别是在stride为1且pad正确设置时特征图尺寸不变的情况。以LeNet-5为例,详细解析了网络参数量的计算,包括卷积层和全连接层的参数。通过计算,得出模型参数量,并与实际Caffe模型的大小进行比较,两者接近,说明了计算的准确性。
摘要由CSDN通过智能技术生成

Draw_convnet

这里写图片描述

这幅图是通过开源的工具draw_convnet(https://github.com/gwding/draw_convnet)生成的。在清楚整个前向计算网络中的每一个层的输入输出以及参数设置后可以自己手动画出计算图出来,对于参数量计算就很直观了。

feature map大小计算

输入:N0*C0*H0*W0 
输出:N1*C1*H1*W1 
输出的feature map大小: 
H1=(H0
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值