Mayurji / Image-Classification-PyTorch
Learning and Building Convolutional Neural Networks using PyTorch
☆207Updated 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…☆84Updated 3 years ago
- A summarization of Transformer-based architectures for CV tasks, including image classification, object detection, segmentation, and Few-…☆108Updated 2 years ago
- EfficientNetV2 implementation using PyTorch☆129Updated 2 years ago
- Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]☆421Updated last month
- Implementation of MobileNet V1, V2, V3☆65Updated 4 years ago
- ☆95Updated 3 years ago
- The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Netwo…☆435Updated 2 years ago
- A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer"☆521Updated 3 years ago
- ☆153Updated 3 years ago
- CBAM: Convolutional Block Attention Module for CIFAR10 on ResNet backbone with Pytorch☆113Updated 5 years ago
- Pytorch implementation(LeNet, VGGNet, GAN, UNet, Mask R-CNN, ...)☆28Updated 3 years ago
- Nearly Perfect & Easily Understandable PyTorch Implementation of SKNet☆94Updated 3 years ago
- A collection of SOTA Image Classification Models in PyTorch☆152Updated 2 years ago
- Machine learning, deep learning, evolution algorithm, the fans of math☆12Updated 3 years ago
- Implementation of Swin Transformer with Pytorch☆111Updated 3 years ago
- Spatial and Channel Attention in CNN Architectures for Image Classification task☆27Updated 4 years ago
- ☆298Updated 3 years ago
- ☆66Updated 3 years ago
- PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" [CVPR 2022].☆192Updated 2 years ago
- EfficientNetV2 pytorch (pytorch lightning) implementation with pretrained model☆68Updated 2 years ago
- Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks☆259Updated 3 years ago
- The official code for our TIP paper 'LayerCAM: Exploring Hierarchical Class Activation Maps for Localization'☆124Updated 3 years ago
- Segmentation for vertebra in MR images☆102Updated 3 years ago
- Implementation of the Swin Transformer in PyTorch.☆821Updated 3 years ago
- Implementation of Conv-based and Vit-based networks designed for CIFAR.☆72Updated 2 years ago
- FcaNet: Frequency Channel Attention Networks☆539Updated 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
- Awesome List of Attention Modules and Plug&Play Modules in Computer Vision☆1,153Updated last year
- Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet☆489Updated last year
- A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder☆158Updated 4 years ago