hysts / pytorch_image_classificationLinks
PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet
☆1,408Updated 3 years ago
Alternatives and similar repositories for pytorch_image_classification
Users that are interested in pytorch_image_classification are comparing it to the libraries listed below
Sorting:
- Classification with PyTorch.☆1,731Updated last year
- experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN a…☆728Updated 6 years ago
- PyTorch implementation of SENet☆2,323Updated 4 years ago
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,556Updated 11 months ago
- Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeN…☆2,673Updated 2 years ago
- Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"☆2,158Updated 2 years ago
- ResNeSt: Split-Attention Networks☆3,256Updated 2 years ago
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆754Updated 4 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,154Updated last year
- Flops counter for neural networks in pytorch framework☆2,904Updated 5 months ago
- Squeeze-and-Excitation Networks☆3,526Updated 6 years ago
- Implementation of Non-local Block.☆1,573Updated 2 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,243Updated 8 months ago
- A memory-efficient implementation of DenseNets☆1,536Updated 2 years ago
- PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)☆795Updated 5 years ago
- Sandbox for training deep learning networks☆3,011Updated 9 months ago
- Count the MACs / FLOPs of your PyTorch model.☆5,006Updated 11 months ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,092Updated 3 years ago
- This implements training of popular model architectures, such as AlexNet, ResNet and VGG on the ImageNet dataset(Now we supported alexnet…☆425Updated 6 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆882Updated 4 years ago
- mobilenetv3 with pytorch，provide pre-train model☆1,739Updated 2 years ago
- 95.47% on CIFAR10 with PyTorch☆6,220Updated 2 years ago
- Official Pytorch implementation of CutMix regularizer☆1,239Updated 4 years ago
- A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility☆1,945Updated 2 years ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,455Updated 6 years ago
- a pytorch code about Residual Attention Network. This code is based on two projects from☆683Updated 6 years ago
- [ICCV 2017] Torch code for Grad-CAM☆1,568Updated 2 years ago
- My best practice of training large dataset using PyTorch.☆1,100Updated last year
- Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS☆1,572Updated last year
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,395Updated 5 years ago