bentrevett / pytorch-image-classification
Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.
☆979Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for pytorch-image-classification
- Some custom dataset examples for PyTorch☆862Updated 4 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,364Updated 2 years ago
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆872Updated last year
- Pytorch implementation of convolutional neural network visualization techniques☆7,869Updated 2 years ago
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,294Updated 4 months ago
- A Pytorch Computer Vision template to quick start your next project! 🚀🚀☆346Updated 3 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,020Updated 8 months ago
- Clean, scalable and easy to use ResNet implementation in Pytorch☆194Updated 4 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,038Updated 2 years ago
- In-depth tutorials for implementing deep learning models on your own with PyTorch.☆1,511Updated 11 months ago
- experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN a…☆716Updated 5 years ago
- pytorch tutorial for beginners☆2,999Updated 2 years ago
- Early stopping for PyTorch☆1,230Updated last week
- Classification with PyTorch.☆1,693Updated 5 months ago
- Official Pytorch implementation of CutMix regularizer☆1,221Updated 4 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,036Updated this week
- Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation☆837Updated 9 months ago
- Image Test Time Augmentation with PyTorch!☆978Updated last year
- View model summaries in PyTorch!☆2,603Updated this week
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆10,601Updated last month
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,183Updated last month
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆2,265Updated 8 months ago
- A PyTorch Implementation of Focal Loss.☆962Updated 5 years ago
- A Simple pytorch implementation of GradCAM and GradCAM++☆359Updated 5 years ago
- An unofficial styleguide and best practices summary for PyTorch☆1,931Updated 2 years ago
- Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.☆2,747Updated last year
- PyTorch deep learning projects made easy.☆4,762Updated 5 months ago
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,520Updated last month
- Siamese and triplet networks with online pair/triplet mining in PyTorch☆3,107Updated last year
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,182Updated 3 years ago