轻量化模型:Mobile Net V1 核心 分组卷积+1*1逐点卷积

参考: https://zhuanlan.zhihu.com/p/127194294

代码实现:

def conv_dw(inp, oup, stride, leaky=0.1):
return nn.Sequential(
nn.Conv2d(inp, inp, 3, stride, 1, groups=inp, bias=False),
nn.BatchNorm2d(inp),
nn.LeakyReLU(negative_slope= leaky,inplace=True),
nn.Conv2d(inp, oup, 1, 1, 0, bias=False),
nn.BatchNorm2d(oup),
nn.LeakyReLU(negative_slope= leaky,inplace=True),
)

注:Conv2d 的 组 和 输入的通道数 \(C_{in}\) 一样, bias = False。

class torch.nn.LeakyReLU(negative_slope=0.01, inplace=False)

参数:

negative_slope-控制负斜率的角度。默认值:1e-2

inplace-可以选择就地执行操作。默认值:False

参考:

https://vimsky.com/examples/usage/python-torch.nn.LeakyReLU-pt.html

https://blog.csdn.net/zhanly19/article/details/99232815

其他可参考:

https://www.cnblogs.com/xiaxuexiaoab/p/12319056.html

posted @   Zenith_Hugh  阅读(47)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示

喜欢请打赏

扫描二维码打赏

微信打赏