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.
☆84Updated 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
- fpn_resnet、resnet-se、resnet-cbam☆33Updated 5 years ago
- ☆298Updated 3 years ago
- CBAM: Convolutional Block Attention Module for CIFAR10 on ResNet backbone with Pytorch☆113Updated 5 years ago
- ☆153Updated 3 years ago
- Self-attention、Non-local、SE、SK、CBAM、DANet☆463Updated 3 years ago
- 基于Swin-transformer训练图像分类并部署web端☆86Updated 2 years ago
- ☆34Updated last year
- Classification with backbone Resnet and attentions: SE-Channel Attention, BAM - (Spatial Attention, Channel Attention, Joint Attention), …☆59Updated 4 years ago
- PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, Google…☆55Updated 9 months ago
- Segmentation for vertebra in MR images☆102Updated 3 years ago
- The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Netwo…☆434Updated 2 years ago
- Gluon implementation of channel-attention modules: SE, ECA, GCT☆39Updated 4 years ago
- (BIBM22) This is the official code repository for "MALUNet: A Muti-Attention and Light-weight UNet for Skin Lesion Segmentation".☆81Updated last year
- Official repository of ACmix (CVPR2022)☆407Updated 2 years ago
- A PyTorch implementation of Inception-v4 and Inception-ResNet-v2.☆114Updated 5 years ago
- This is a pytorch project about medical segmentation☆57Updated 3 years ago
- pytorch版—使用resnet50迁移学习实现皮肤病图片的二分类☆125Updated 5 years ago
- This repo holds the code of TransFuse: Fusing Transformers and CNNs for Medical Image Segmentation☆193Updated 2 years ago
- 这里面存放了一些目标检测算法的数据增强方法。如mosaic、mixup。☆159Updated 2 years ago
- Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]☆421Updated last month
- 基于Pytorch框架的语义分割学习,FCN,UNet,DeepLab 基于VOC数据集。☆46Updated 2 years ago
- ☆35Updated last year
- CAM图的resnet50版本☆152Updated 4 years ago
- This repository contains code for a multiple classification image segmentation model based on UNet and UNet++☆168Updated 4 years ago
- 这个是利用pytorch中的torchvision实现的一个maskrcnn的目标检测和实例分割的小例子☆110Updated 4 years ago
- SSD 改进:backbone增加resnet,并针对resnet加入attention机制,为了充分利用细节特征和语义特征,还加入了特征融合模型☆75Updated 4 years ago
- [pytorch] DANet: Dual Attention Network for Scene Segmentation☆71Updated 5 years ago
- CAM', 'ScoreCAM', 'SSCAM', 'ISCAM' 'GradCAM', 'GradCAMpp', 'SmoothGradCAMpp', 'XGradCAM', 'LayerCAM' using by PyTorch.☆72Updated 3 years ago
- CNN可视化代码,帮助了解建立GradCam过程, 代码中文注解☆101Updated 5 years ago
- 可视化神经网络☆39Updated 2 years ago