search

xiaolai-sqlai / mobilenetv3

mobilenetv3 with pytorch，provide pre-train model
MIT License
1.6k stars 340 forks source link

Mobile-Net #20

Open chuliuT opened 4 years ago

chuliuT commented 4 years ago

@xiaolai-sqlai : fun: class MobileNetV3_Large(nn.Module): self.bneck = nn.Sequential( .... Block(5, 112, 672, 160, hswish(), SeModule(160), 1), Block(5, 160, 672, 160, hswish(), SeModule(160), 2), Block(5, 160, 960, 160, hswish(), SeModule(160), 1), ) paper define: Block(5, 112, 672, 160, hswish(), SeModule(160), 2), Block(5, 160, 672, 160, hswish(), SeModule(160), 1), Block(5, 160, 960, 160, hswish(), SeModule(160), 1),

HuColin commented 4 years ago

https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_v3.py#L490-L492 开源的的代码是

        mbv3_op_se(ef=6, n=112, k=3, s=1, act=hard_swish),
        mbv3_op_se(ef=6, n=160, k=5, s=2, act=hard_swish),
        mbv3_op_se(ef=6, n=160, k=5, s=1, act=hard_swish),