ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design
论文链接:https://arxiv.org/pdf/1807.11164.pdf
PyTorch:https://github.com/shanglianlm0525/PyTorch-Networks
ShuffleNet V2 unit
ShuffleNet v2 architecture
PyTorch代码
import torch
import torch.nn as nn
import torchvision
def Conv3x3BNReLU(in_channels,out_channels,stride,groups):
return nn.Sequential(
nn.Conv2d(in_channels=in_channels,