aaron-xichen / pytorch-playground
Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
☆2,627Updated last year
Related projects ⓘ
Alternatives and complementary repositories for pytorch-playground
- Classification with PyTorch.☆1,693Updated 5 months ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,038Updated 2 years ago
- Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…☆4,294Updated 4 months ago
- 95.47% on CIFAR10 with PyTorch☆5,992Updated last year
- Deformable Convolutional Networks☆4,050Updated 3 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,364Updated 2 years ago
- A memory-efficient implementation of DenseNets☆1,521Updated last year
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,444Updated 6 years ago
- Squeeze-and-Excitation Networks☆3,397Updated 5 years ago
- PyTorch implementation of SENet☆2,285Updated 3 years ago
- Class Activation Mapping☆1,844Updated 2 years ago
- Semantic Segmentation Architectures Implemented in PyTorch☆3,393Updated last year
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,871Updated this week
- Faster RCNN with PyTorch☆1,725Updated 3 years ago
- Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.☆1,225Updated 5 months ago
- Pytorch implementation of convolutional neural network visualization techniques☆7,869Updated 2 years ago
- PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)☆1,748Updated last year
- A faster pytorch implementation of faster r-cnn☆7,695Updated 2 years ago
- pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)☆1,817Updated 4 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,020Updated 8 months ago
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,821Updated 5 years ago
- Count the MACs / FLOPs of your PyTorch model.☆4,888Updated 4 months ago
- A PyTorch implementation of MobileNet V2 architecture and pretrained model.☆1,376Updated 5 years ago
- A PyTorch implementation of DenseNet.☆832Updated 6 years ago
- Non-local Neural Networks for Video Classification☆1,975Updated 3 years ago
- [ICCV 2017] Torch code for Grad-CAM☆1,501Updated 2 years ago
- pytorch tutorial for beginners☆2,999Updated 2 years ago
- Collection of generative models in Pytorch version.☆2,601Updated 4 years ago
- A CV toolkit for my papers.☆2,041Updated 5 months ago
- YOLOv2 in PyTorch☆1,550Updated 3 years ago