Mayurji / Image-Classification-PyTorchLinks
Learning and Building Convolutional Neural Networks using PyTorch
☆213Updated 3 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
Sorting:
- Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https…☆92Updated 3 years ago
- CBAM: Convolutional Block Attention Module for CIFAR10 on ResNet backbone with Pytorch☆118Updated 5 years ago
- A collection of SOTA Image Classification Models in PyTorch☆158Updated 3 years ago
- EfficientNetV2 implementation using PyTorch☆129Updated 2 years ago
- ☆304Updated 3 years ago
- Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]☆424Updated 4 months ago
- ☆97Updated 4 years ago
- EfficientNetV2 pytorch (pytorch lightning) implementation with pretrained model☆74Updated 2 years ago
- Implementation of Swin Transformer with Pytorch☆112Updated 4 years ago
- A summarization of Transformer-based architectures for CV tasks, including image classification, object detection, segmentation, and Few-…☆111Updated 3 years ago
- Unofficial Implementation of ECANets (CVPR 2020) for the Reproducibility Challenge 2020.☆31Updated 3 years ago
- Spatial and Channel Attention in CNN Architectures for Image Classification task☆27Updated 4 years ago
- ☆154Updated 3 years ago
- A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer"☆537Updated 3 years ago
- PyTorch implementation of EfficientNetV2 family☆477Updated 3 years ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆257Updated last year
- A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"☆388Updated 3 years ago
- Implementation of MobileNet V1, V2, V3☆71Updated 4 years ago
- PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, Google…☆55Updated last year
- The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Netwo…☆444Updated 2 years ago
- Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet☆498Updated last year
- pytorch版—使用resnet50迁移学习实现皮肤病图片的二分类☆125Updated 5 years ago
- Nearly Perfect & Easily Understandable PyTorch Implementation of SKNet☆97Updated 3 years ago
- FcaNet: Frequency Channel Attention Networks☆553Updated 4 years ago
- Pytorch implementation(LeNet, VGGNet, GAN, UNet, Mask R-CNN, ...)☆28Updated 4 years ago
- Restore the official code 100% and improve it to make it easier to use.☆50Updated 2 years ago
- Multi-label classification based on timm.☆119Updated 3 years ago
- Segmentation for vertebra in MR images☆108Updated 3 years ago
- Gluon implementation of channel-attention modules: SE, ECA, GCT☆41Updated 4 years ago
- ☆72Updated 4 years ago