CBAM.PyTorch

Feature map refiner

PyTorch implementation of the CBAM module for refining feature maps in deep networks

Non-official implement of Paper:CBAM: Convolutional Block Attention Module

GitHub

1k stars
12 watching
285 forks
Language: Python
last commit: over 1 year ago
Linked from 1 awesome list


Backlinks from these awesome lists:

Related projects:

Repository Description Stars
bmsookim/wide-resnet.pytorch PyTorch implementation of wide residual networks for image classification tasks on CIFAR-10 and CIFAR-100 datasets 461
rentruewang/koila A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution 1,821
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
tengshaofeng/residualattentionnetwork-pytorch Implementing a deep learning framework for image classification using Residual Attention Network architecture 680
minglllli/cbafed A PyTorch implementation of a method for improving semi-supervised learning in federated settings by adapting pseudo labels to balance classes. 7
prabhuomkar/pytorch-cpp A C++ implementation of PyTorch tutorials 1,965
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
randl/shufflenetv2-pytorch An implementation of a lightweight convolutional neural network architecture for mobile devices 191
oyam/pytorch-dpns PyTorch implementation of a deep learning model for image segmentation 90
nearai/torchfold A PyTorch module for dynamic batching and optimized computation on deep neural networks 221
lemonhu/ner-bert-pytorch A PyTorch implementation of named entity recognition using the Google AI's pre-trained BERT model for Chinese text data. 438
pp00704831/banet-tip-2022 A PyTorch implementation of an attention network for dynamic scene deblurring 37
bamos/densenet.pytorch An implementation of a deep neural network architecture in PyTorch 832
l0sg/relational-rnn-pytorch An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling 244