Mayurji / Image-Classification-PyTorch
Learning and Building Convolutional Neural Networks using PyTorch
☆203Updated 2 years ago
Alternatives and similar repositories for Image-Classification-PyTorch:
Users that are interested in Image-Classification-PyTorch are comparing it to the libraries listed below
- Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https…☆81Updated 3 years ago
- Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]☆417Updated 2 weeks ago
- EfficientNetV2 implementation using PyTorch☆125Updated 2 years ago
- Spatial and Channel Attention in CNN Architectures for Image Classification task☆27Updated 4 years ago
- A collection of SOTA Image Classification Models in PyTorch☆146Updated 2 years ago
- Implementation of Swin Transformer with Pytorch☆109Updated 3 years ago
- CBAM: Convolutional Block Attention Module for CIFAR10 on ResNet backbone with Pytorch☆108Updated 5 years ago
- A summarization of Transformer-based architectures for CV tasks, including image classification, object detection, segmentation, and Few-…☆107Updated 2 years ago
- ☆295Updated 3 years ago
- The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Netwo…☆419Updated 2 years ago
- Pytorch implementation for Semantic Segmentation with multi models (Deeplabv3, Deeplabv3_plus, PSPNet, UNet, UNet_AutoEncoder, UNet_neste…☆191Updated 4 years ago
- FcaNet: Frequency Channel Attention Networks☆526Updated 3 years ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆242Updated last year
- PyTorch implementation of EfficientNetV2 family☆465Updated 2 years ago
- ☆94Updated 3 years ago
- ☆153Updated 3 years ago
- Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks☆255Updated 3 years ago
- Restore the official code 100% and improve it to make it easier to use.☆48Updated last year
- Pytorch implementation(LeNet, VGGNet, GAN, UNet, Mask R-CNN, ...)☆27Updated 3 years ago
- Awesome List of Attention Modules and Plug&Play Modules in Computer Vision☆1,132Updated last year
- EfficientNetV2 pytorch (pytorch lightning) implementation with pretrained model☆65Updated 2 years ago
- Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet☆482Updated last year
- Implementation of MobileNet V1, V2, V3☆63Updated 4 years ago
- ☆145Updated 2 months ago
- Collection of different Unet Variant suchas VggUnet, ResUnet, DenseUnet, Unet. AttUnet, MobileNetUnet, NestedUNet, R2AttUNet, R2UNet, SEU…☆97Updated 2 years ago
- CNN经典网络结构复现：LeNet5、AlexNet、VGG、ResNet、InceptionNet等。☆85Updated 3 years ago
- Self-attention、Non-local、SE、SK、CBAM、DANet☆459Updated 3 years ago
- A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer"☆509Updated 3 years ago
- [ICCV 2021] Code for approximated exponential maximum pooling☆292Updated last year
- Implementing ConvNext in PyTorch☆65Updated 2 years ago