Knight825 / models-pytorch

This is a torchvision style CNN models collection based on pytorch.
Apache License 2.0
20 stars 5 forks source link
attention-mechanism cbam cbam-resnet cnn-model crossstagepartial csp-densenet csp-resnet exactfusionmodel pytorch se-resnet self-attention senet

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