aaron-xichen / pytorch-playground
Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
☆2,624Updated last year
Related projects ⓘ
Alternatives and complementary repositories for pytorch-playground
- Classification with PyTorch.☆1,690Updated 4 months ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,030Updated 2 years ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,444Updated 6 years ago
- A memory-efficient implementation of DenseNets☆1,522Updated last year
- Deformable Convolutional Networks☆4,047Updated 3 years ago
- Squeeze-and-Excitation Networks☆3,389Updated 5 years ago
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,375Updated 5 years ago
- Count the MACs / FLOPs of your PyTorch model.☆4,875Updated 4 months ago
- 95.47% on CIFAR10 with PyTorch☆5,983Updated last year
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,870Updated 2 months ago
- Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.☆1,222Updated 4 months ago
- PyTorch implementation of SENet☆2,283Updated 3 years ago
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,272Updated 3 months ago
- A CV toolkit for my papers.☆2,040Updated 5 months ago
- Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts☆2,294Updated 2 years ago
- Model analyzer in PyTorch☆1,466Updated last year
- A PyTorch implementation of DenseNet.☆830Updated 6 years ago
- Faster RCNN with PyTorch☆1,722Updated 3 years ago
- Fine-tune pretrained Convolutional Neural Networks with PyTorch☆724Updated 2 months ago
- A simplified implemention of Faster R-CNN that replicate performance from origin paper☆3,988Updated 3 years ago
- PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)☆1,745Updated last year
- YOLOv2 in PyTorch☆1,548Updated 3 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,016Updated 8 months ago
- Awesome Knowledge Distillation☆3,466Updated 3 weeks ago
- ResNeSt: Split-Attention Networks☆3,236Updated last year
- A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility☆1,859Updated last year
- Pytorch implementation of convolutional neural network visualization techniques☆7,859Updated 2 years ago
- Non-local Neural Networks for Video Classification☆1,976Updated 3 years ago
- pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)☆1,817Updated 3 years ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,820Updated 5 years ago