您好,欢迎来到步遥情感网。
搜索
您的当前位置:首页关于mobilenet中提出的depthwise convolution的解析及pytorch实现

关于mobilenet中提出的depthwise convolution的解析及pytorch实现

来源:步遥情感网

本人觉得很多博客都没有解释清楚,或者存在或多或少的错误。

我个人觉得,以下这篇博文讲得很清晰易懂:


以下是pytorch代码:

class depthwise_conv2d(nn.Module):
    def __init__(self, n_in, n_out):
        super(depthwise_separable_conv, self).__init__()
        self.depth_wise = nn.Conv2d(n_in, n_in, kernel_size=3, padding=1, groups=n_in)
        self.point_wise = nn.Conv2d(n_in, n_out, kernel_size=1)

    def forward(self, x):
        out = self.depth_wise(x)
        out = self.point_wise(out)
        return out

 

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- obuygou.com 版权所有 赣ICP备2024042798号-5

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务