models-pytorch

This is a torchvision style CNN models collection based on pytorch.

requirement: python 3.5+, pytorch1.1+,torchvision0.4+

Module introduction: 1.attention module: 1.1 Squeeze and Excitation module:https://arxiv.org/pdf/1709.01507.pdf; 1.2 Convolutional Block Attention Module:https://arxiv.org/pdf/1807.06521.pdf;

2.cross stage partial module: 2.1 cross stage partial module(csp) and Exact Fusion Model(efm):https://arxiv.org/pdf/1911.11929.pdf