aaron-xichen / pytorch-playground
Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
☆2,664Updated 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,724Updated 10 months ago
- A memory-efficient implementation of DenseNets☆1,532Updated last year
- Squeeze-and-Excitation Networks☆3,491Updated 6 years ago
- Deformable Convolutional Networks☆4,074Updated 3 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,403Updated 3 years ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,455Updated 6 years ago
- Faster RCNN with PyTorch☆1,741Updated 3 years ago
- PyTorch implementation of SENet☆2,319Updated 4 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,076Updated 3 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,932Updated this week
- PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)☆1,766Updated 2 years ago
- Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts☆2,323Updated 2 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,823Updated 5 years ago
- 95.47% on CIFAR10 with PyTorch☆6,158Updated 2 years ago
- A PyTorch implementation of DenseNet.☆840Updated 6 years ago
- YOLOv2 in PyTorch☆1,561Updated 3 years ago
- pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)☆1,819Updated 4 years ago
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,392Updated 5 years ago
- Siamese and triplet networks with online pair/triplet mining in PyTorch☆3,136Updated last year
- A CV toolkit for my papers.☆2,048Updated 4 months ago
- Fully Convolutional Networks for Semantic Segmentation by Jonathan Long*, Evan Shelhamer*, and Trevor Darrell. CVPR 2015 and PAMI 2016.☆3,333Updated 3 years ago
- ☆3,221Updated 6 years ago
- Class Activation Mapping☆1,860Updated 2 years ago
- Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).☆4,798Updated last year
- Fine-tune pretrained Convolutional Neural Networks with PyTorch☆722Updated 8 months ago
- Non-local Neural Networks for Video Classification☆1,980Updated 3 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,500Updated 4 years ago
- Pytorch implementation of convolutional neural network visualization techniques☆8,006Updated 3 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,038Updated last year
- A lightweight library for PyTorch training tools and utilities☆1,688Updated 2 weeks ago