Ugenteraan / ResNet-50-CBAM-PyTorch
Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https://www.kaggle.com/puneet6060/intel-image-classification.
☆87Updated 3 years ago
Alternatives and similar repositories for ResNet-50-CBAM-PyTorch:
Users that are interested in ResNet-50-CBAM-PyTorch are comparing it to the libraries listed below
- ☆299Updated 3 years ago
- fpn_resnet、resnet-se、resnet-cbam☆33Updated 5 years ago
- CBAM: Convolutional Block Attention Module for CIFAR10 on ResNet backbone with Pytorch☆116Updated 5 years ago
- Gluon implementation of channel-attention modules: SE, ECA, GCT☆40Updated 4 years ago
- 基于Swin-transformer训练图像分类并部署web端☆90Updated 2 years ago
- The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Netwo…☆439Updated 2 years ago
- Self-attention、Non-local、SE、SK、CBAM、DANet☆473Updated 3 years ago
- ☆153Updated 3 years ago
- Classification with backbone Resnet and attentions: SE-Channel Attention, BAM - (Spatial Attention, Channel Attention, Joint Attention), …☆59Updated 4 years ago
- 多模态数据融合:为了完成多模态数据融合,首先利用VGG16网络和cifar10数据集完成多输入网络的分类,在VGG16的基础之上,将前三层特征提取网络作为不同输入的特征提取网络,在中间层进行特征拼接,后面的卷积层用于提取融合特征,最后加上全连接层。该网络稍作修改就能同时提取…☆90Updated 4 years ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆252Updated last year
- Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks☆261Updated 3 years ago
- ☆96Updated 3 years ago
- Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]☆421Updated 2 months ago
- 可视化神经网络☆38Updated 2 years ago
- 可视化特征图教程☆74Updated 3 years ago
- ☆36Updated last year
- CAM', 'ScoreCAM', 'SSCAM', 'ISCAM' 'GradCAM', 'GradCAMpp', 'SmoothGradCAMpp', 'XGradCAM', 'LayerCAM' using by PyTorch.☆73Updated 3 years ago
- PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, Google…☆55Updated 10 months ago
- pytorch resnet image-classification☆19Updated 2 years ago
- SSD 改进:backbone增加resnet,并针对resnet加入attention机制,为了充分利用细节特征和语义特征,还加入了特征融合模型☆78Updated 4 years ago
- [pytorch] DANet: Dual Attention Network for Scene Segmentation☆71Updated 5 years ago
- Segmentation for vertebra in MR images☆104Updated 3 years ago
- DA-TransUNet: Combining Dual Attention of Position and Channel with Transformer U-net for Medical Image Segmentation☆150Updated 6 months ago
- code and trained models for "Attentional Feature Fusion"☆776Updated 3 years ago
- Official repository of ACmix (CVPR2022)☆412Updated 3 years ago
- Nearly Perfect & Easily Understandable PyTorch Implementation of SKNet☆95Updated 3 years ago
- This repo holds the code of TransFuse: Fusing Transformers and CNNs for Medical Image Segmentation☆197Updated 2 years ago
- image classifier implement in pytoch.☆112Updated 2 years ago
- FcaNet: Frequency Channel Attention Networks☆278Updated 4 years ago