This is a torchvision style CNN models collection based on pytorch.
pytorch
attention-mechanism
cnn-model
senet
self-attention
cbam
cbam-resnet
se-resnet
crossstagepartial
csp-resnet
csp-densenet
exactfusionmodel
-
Updated
Sep 15, 2020 - Python