aaron-xichen / pytorch-playground
Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
☆2,640Updated 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,706Updated 7 months ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,051Updated 2 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,378Updated 3 years ago
- Deformable Convolutional Networks☆4,064Updated 3 years ago
- A memory-efficient implementation of DenseNets☆1,527Updated last year
- 95.47% on CIFAR10 with PyTorch☆6,065Updated last year
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,450Updated 6 years ago
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,379Updated 6 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,028Updated 10 months ago
- Faster RCNN with PyTorch☆1,724Updated 3 years ago
- YOLOv2 in PyTorch☆1,555Updated 3 years ago
- pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)☆1,817Updated 4 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,822Updated 5 years ago
- A PyTorch implementation of DenseNet.☆835Updated 6 years ago
- A faster pytorch implementation of faster r-cnn☆7,733Updated 2 years ago
- PyTorch implementation of SENet☆2,298Updated 3 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,904Updated 2 weeks ago
- Pytorch implementation of convolutional neural network visualization techniques☆7,939Updated 3 weeks ago
- PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)☆1,756Updated last year
- Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).☆4,764Updated last year
- A simplified implemention of Faster R-CNN that replicate performance from origin paper☆4,010Updated 3 years ago
- Count the MACs / FLOPs of your PyTorch model.☆4,945Updated 6 months ago
- Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts☆2,311Updated 2 years ago
- Squeeze-and-Excitation Networks☆3,435Updated 5 years ago
- A PyTorch Implementation of Single Shot MultiBox Detector☆5,171Updated 3 years ago
- Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.☆1,250Updated 7 months ago
- Class Activation Mapping☆1,853Updated 2 years ago
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,385Updated 5 years ago
- 3.8% and 18.3% on CIFAR-10 and CIFAR-100☆1,297Updated 5 years ago
- PyTorch implementation of CNNs for CIFAR benchmark☆705Updated 3 years ago