Mayurji / Image-Classification-PyTorch
Learning and Building Convolutional Neural Networks using PyTorch
☆212Updated 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…☆89Updated 3 years ago
- CBAM: Convolutional Block Attention Module for CIFAR10 on ResNet backbone with Pytorch☆117Updated 5 years ago
- CNN经典网络结构复现：LeNet5、AlexNet、VGG、ResNet、InceptionNet等。☆94Updated 4 years ago
- Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]☆422Updated 3 months ago
- Unofficial implementations of some classical CNN backbones with pytorch☆169Updated 3 years ago
- EfficientNetV2 implementation using PyTorch☆129Updated 2 years ago
- Pytorch implementation(LeNet, VGGNet, GAN, UNet, Mask R-CNN, ...)☆28Updated 3 years ago
- ☆300Updated 3 years ago
- Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks☆261Updated 3 years ago
- ☆153Updated 3 years ago
- PyTorch implementation of EfficientNetV2 family☆472Updated 3 years ago
- Implementation of MobileNet V1, V2, V3☆69Updated 4 years ago
- Nearly Perfect & Easily Understandable PyTorch Implementation of SKNet☆96Updated 3 years ago
- The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Netwo…☆440Updated 2 years ago
- A PyTorch implementation of VGG16. This could be considered as a variant of the original VGG16 since BN layers are added after each conv…☆186Updated 5 years ago
- Restore the official code 100% and improve it to make it easier to use.☆50Updated last year
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆256Updated last year
- ☆96Updated 3 years ago
- Self-attention、Non-local、SE、SK、CBAM、DANet☆477Updated 3 years ago
- Semantic Segmentation in Pytorch. Network include: FCN、FCN_ResNet、SegNet、UNet、BiSeNet、BiSeNetV2、PSPNet、DeepLabv3_plus、 HRNet、DDRNet☆494Updated last year
- FcaNet: Frequency Channel Attention Networks☆542Updated 4 years ago
- ⛵️ Implementation a variety of popular Image Classification Models using TensorFlow2. [ResNet, GoogLeNet, VGG, Inception-v3, Inception-v4…☆105Updated 5 years ago
- ☆21Updated 3 years ago
- Awesome List of Attention Modules and Plug&Play Modules in Computer Vision☆1,182Updated 2 years ago
- VGG16 Net implementation from PyTorch Examples scripts for ImageNet dataset☆87Updated 4 years ago
- A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer"☆530Updated 3 years ago
- Implementation of the Swin Transformer in PyTorch.☆827Updated 4 years ago
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆749Updated 4 years ago
- A Simple pytorch implementation of GradCAM and GradCAM++☆378Updated 6 years ago
- The official code for our TIP paper 'LayerCAM: Exploring Hierarchical Class Activation Maps for Localization'☆126Updated 3 years ago