bentrevett / pytorch-image-classificationLinks
Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.
☆1,068Updated 4 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:
- Clean, scalable and easy to use ResNet implementation in Pytorch☆217Updated 5 years ago
- Some custom dataset examples for PyTorch☆874Updated 5 years ago
- In-depth tutorials for implementing deep learning models on your own with PyTorch.☆1,563Updated 2 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,438Updated 4 years ago
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,215Updated 4 years ago
- PyTorch Computer Vision Cookbook, Published by Packt☆313Updated last month
- Early stopping for PyTorch☆1,270Updated last year
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆914Updated 2 years ago
- Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation☆880Updated last year
- A Pytorch Computer Vision template to quick start your next project! 🚀🚀☆360Updated last year
- Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)☆2,120Updated 3 years ago
- Paper implementations from scratch and machine learning tutorials☆348Updated 2 years ago
- My assignment solutions for CS231n - Convolutional Neural Networks for Visual Recognition☆488Updated 3 years 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,285Updated last month
- Augmentations usage examples for albumentations library☆445Updated 3 months ago
- SSD: Single Shot MultiBox Detector | a PyTorch Tutorial to Object Detection☆3,121Updated 2 years ago
- SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]☆1,458Updated 2 years ago
- Vision Transformer (ViT) in PyTorch☆845Updated 3 years ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆2,483Updated last year
- 95.47% on CIFAR10 with PyTorch☆6,356Updated 2 years ago
- experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN a…☆743Updated 6 years ago
- A Generative Adversarial Networks (GANs) Specialization made by deeplearning.ai on Coursera☆161Updated 5 years ago
- Pytorch Implementation of AlexNet☆214Updated 2 years ago
- Semantic segmentation models, datasets and losses implemented in PyTorch.☆1,810Updated 11 months ago
- A curated list of deep learning image classification papers and codes☆3,051Updated 3 years ago
- Official Pytorch implementation of CutMix regularizer☆1,248Updated 5 years ago
- PyTorch Tutorials from my YouTube channel☆1,895Updated 2 years ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,327Updated last month
- mixup: Beyond Empirical Risk Minimization☆1,199Updated 4 years ago
- High-quality Neural Networks for Computer Vision 😎☆446Updated 3 years ago