aaron-xichen / pytorch-playground
Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
☆2,655Updated 2 years ago
Alternatives and similar repositories for pytorch-playground:
Users that are interested in pytorch-playground are comparing it to the libraries listed below
- Classification with PyTorch.☆1,719Updated 9 months ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,452Updated 6 years ago
- A memory-efficient implementation of DenseNets☆1,529Updated last year
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,458Updated 8 months ago
- Deformable Convolutional Networks☆4,069Updated 3 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,052Updated 2 years ago
- PyTorch implementation of SENet☆2,315Updated 4 years ago
- 95.47% on CIFAR10 with PyTorch☆6,115Updated 2 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,389Updated 3 years ago
- Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts☆2,316Updated 2 years ago
- Squeeze-and-Excitation Networks☆3,473Updated 6 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,033Updated last year
- Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).☆4,787Updated last year
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,389Updated 5 years ago
- Class Activation Mapping☆1,854Updated 2 years ago
- A PyTorch implementation of DenseNet.☆836Updated 6 years ago
- ResNeSt: Split-Attention Networks☆3,246Updated 2 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,919Updated 2 months ago
- Count the MACs / FLOPs of your PyTorch model.☆4,964Updated 8 months ago
- Model analyzer in PyTorch☆1,477Updated 2 years ago
- PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)☆1,762Updated 2 years ago
- YOLOv2 in PyTorch☆1,558Updated 3 years ago
- Pytorch implementation of convolutional neural network visualization techniques☆7,978Updated 2 months ago
- Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.☆1,266Updated 9 months ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,480Updated last year
- mobilenetv3 with pytorch,provide pre-train model☆1,702Updated last year
- Non-local Neural Networks for Video Classification☆1,977Updated 3 years ago
- A PyTorch Implementation of Single Shot MultiBox Detector☆5,183Updated 3 years ago
- Faster RCNN with PyTorch☆1,738Updated 3 years ago
- pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)☆1,820Updated 4 years ago