bentrevett / pytorch-image-classificationLinks
Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.
☆1,023Updated 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:
- Some custom dataset examples for PyTorch☆867Updated 5 years ago
- Clean, scalable and easy to use ResNet implementation in Pytorch☆209Updated 5 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,364Updated 5 months ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,404Updated 3 years ago
- In-depth tutorials for implementing deep learning models on your own with PyTorch.☆1,541Updated last year
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,044Updated last year
- Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)☆2,036Updated 2 years ago
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,210Updated 3 years ago
- Official Pytorch implementation of CutMix regularizer☆1,239Updated 4 years ago
- Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation☆860Updated last year
- pytorch and cifar10☆215Updated 4 years ago
- Basic implementation of ResNet 50, 101, 152 in PyTorch☆106Updated 3 years ago
- mixup: Beyond Empirical Risk Minimization☆1,180Updated 3 years ago
- 95.47% on CIFAR10 with PyTorch☆6,196Updated 2 years ago
- Paper implementations from scratch and machine learning tutorials☆349Updated last year
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,538Updated 10 months ago
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆904Updated last year
- Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.☆1,280Updated 11 months ago
- Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-…☆2,195Updated this week
- Semantic segmentation models, datasets and losses implemented in PyTorch.☆1,773Updated 3 months ago
- View model summaries in PyTorch!☆2,792Updated last week
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,833Updated last year
- A learning rate range test implementation in PyTorch☆965Updated last month
- Image Test Time Augmentation with PyTorch!☆1,013Updated last year
- A collection of loss functions for medical image segmentation☆3,916Updated last year
- Classification with PyTorch.☆1,731Updated 11 months ago
- A clean and readable Pytorch implementation of CycleGAN☆1,298Updated 3 years ago
- Early stopping for PyTorch☆1,253Updated 6 months ago
- 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
- A smaller subset of 10 easily classified classes from Imagenet, and a little more French☆1,036Updated 2 years ago