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: about 3 years agoLinked from 1 awesome list
weiaicunzai/awesome-image-classificationRelated 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 |