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
1k stars
12 watching
285 forks
Language: Python
last commit: over 1 year ago
Linked from 1 awesome list
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 | 462 |
rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,823 |
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,044 |
tengshaofeng/residualattentionnetwork-pytorch | Implementing a deep learning framework for image classification using Residual Attention Network architecture | 682 |
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,978 |
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. | 442 |
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 | 833 |
l0sg/relational-rnn-pytorch | An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling | 245 |