Ugenteraan / ResNet-50-CBAM-PyTorchLinks
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.
☆97Updated 4 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
Sorting:
- pytorch版—使用resnet50迁移学习实现皮肤病图片的二分 类☆135Updated 6 years ago
- CBAM: Convolutional Block Attention Module for CIFAR10 on ResNet backbone with Pytorch☆130Updated 6 years ago
- ☆307Updated 4 years ago
- Learning and Building Convolutional Neural Networks using PyTorch☆218Updated 3 years ago
- 基于Swin-transformer训练图像分类并部署web端☆94Updated 3 years ago
- fpn_resnet、resnet-se、resnet-cbam☆33Updated 5 years ago
- A PyTorch implementation of Inception-v4 and Inception-ResNet-v2.☆122Updated 6 years ago
- ☆157Updated 4 years ago
- Self-attention、Non-local、SE、SK、CBAM、DANet☆493Updated 4 years ago
- The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Netwo…☆471Updated 3 years ago
- 使用PyTorch实现Unet图像分割☆174Updated 6 years ago
- Classification with backbone Resnet and attentions: SE-Channel Attention, BAM - (Spatial Attention, Channel Attention, Joint Attention), …☆63Updated 5 years ago
- Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]☆439Updated 11 months ago
- Pytorch implementation for Semantic Segmentation with multi models (Deeplabv3, Deeplabv3_plus, PSPNet, UNet, UNet_AutoEncoder, UNet_neste…☆209Updated 5 years ago
- ☆96Updated 4 years ago
- This repository contains code for a multiple classification image segmentation model based on UNet and UNet++☆176Updated 5 years ago
- Gluon implementation of channel-attention modules: SE, ECA, GCT☆42Updated 5 years ago
- Multi-label classification based on timm.☆128Updated 4 years ago
- ☆66Updated 5 years ago
- code and trained models for "Attentional Feature Fusion"☆807Updated 4 years ago
- 基于PyTorch框架实现的图像分类网络☆85Updated 5 years ago
- a baseline for defect segmentation☆30Updated 5 years ago
- PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, Google…☆57Updated last year
- UNet3+/ UNet++/UNet, used in Deep Automatic Portrait Matting in Pytorth☆269Updated 2 years ago
- SSD 改进:backbone增加resnet,并针对resnet加入attention机制,为了充分利用细节特征和语义特征,还加入了特征融合模型☆81Updated 5 years ago
- Keywords: CNN, Fully connect neural network, SFEW dataset, Image Preprocessing, Data Augmentation, Leakey ReLU, k-fold cross validation, …☆14Updated 5 years ago
- Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks☆265Updated 4 years ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆257Updated 3 months ago
- Implementation of our AAAI'22 work: 'UCTransNet: Rethinking the Skip Connections in U-Net from a Channel-wise Perspective with Transforme…☆431Updated last year
- Official repository of ACmix (CVPR2022)☆411Updated 3 years ago