bearpaw / Pytorch Classification
Licence: mit
Classification with PyTorch.
Stars: ✭ 1,268
Programming Languages
python
139335 projects - #7 most used programming language
Projects that are alternatives of or similar to Pytorch Classification
Pytorch Speech Commands
Speech commands recognition with PyTorch
Stars: ✭ 128 (-89.91%)
Mutual labels: classification, resnet, densenet, cifar10, resnext
Classification models
Classification models trained on ImageNet. Keras.
Stars: ✭ 938 (-26.03%)
Mutual labels: resnet, imagenet, densenet, resnext
Caffe Model
Caffe models (including classification, detection and segmentation) and deploy files for famouse networks
Stars: ✭ 1,258 (-0.79%)
Mutual labels: classification, resnet, imagenet, resnext
simpleAICV-pytorch-ImageNet-COCO-training
SimpleAICV:pytorch training example on ImageNet(ILSVRC2012)/COCO2017/VOC2007+2012 datasets.Include ResNet/DarkNet/RetinaNet/FCOS/CenterNet/TTFNet/YOLOv3/YOLOv4/YOLOv5/YOLOX.
Stars: ✭ 276 (-78.23%)
Mutual labels: classification, imagenet, resnet
Keras Idiomatic Programmer
Books, Presentations, Workshops, Notebook Labs, and Model Zoo for Software Engineers and Data Scientists wanting to learn the TF.Keras Machine Learning framework
Stars: ✭ 720 (-43.22%)
Mutual labels: resnet, densenet, resnext
Pytorch Imagenet Cifar Coco Voc Training
Training examples and results for ImageNet(ILSVRC2012)/CIFAR100/COCO2017/VOC2007+VOC2012 datasets.Image Classification/Object Detection.Include ResNet/EfficientNet/VovNet/DarkNet/RegNet/RetinaNet/FCOS/CenterNet/YOLOv3.
Stars: ✭ 130 (-89.75%)
Mutual labels: classification, resnet, imagenet
pytorch2keras
PyTorch to Keras model convertor
Stars: ✭ 788 (-37.85%)
Mutual labels: imagenet, densenet, resnet
Segmentation models
Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
Stars: ✭ 3,575 (+181.94%)
Mutual labels: resnet, densenet, resnext
Cifar Zoo
PyTorch implementation of CNNs for CIFAR benchmark
Stars: ✭ 584 (-53.94%)
Mutual labels: resnet, densenet, resnext
Segmentationcpp
A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.
Stars: ✭ 49 (-96.14%)
Mutual labels: resnet, imagenet, resnext
Basic cnns tensorflow2
A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).
Stars: ✭ 374 (-70.5%)
Mutual labels: resnet, densenet, resnext
Pretrained Models.pytorch
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
Stars: ✭ 8,318 (+555.99%)
Mutual labels: resnet, imagenet, resnext
Pytorch2keras
PyTorch to Keras model convertor
Stars: ✭ 676 (-46.69%)
Mutual labels: resnet, imagenet, densenet
Pytorch Cifar100
Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet, NasNet, Residual Attention Network, SENet, WideResNet)
Stars: ✭ 2,423 (+91.09%)
Mutual labels: resnet, densenet, resnext
Chainer Cifar10
Various CNN models for CIFAR10 with Chainer
Stars: ✭ 134 (-89.43%)
Mutual labels: resnet, densenet, cifar10
Imagenet
This implements training of popular model architectures, such as AlexNet, ResNet and VGG on the ImageNet dataset(Now we supported alexnet, vgg, resnet, squeezenet, densenet)
Stars: ✭ 126 (-90.06%)
Mutual labels: resnet, imagenet, densenet
Tianchi Medical Lungtumordetect
天池医疗AI大赛[第一季]：肺部结节智能诊断 UNet/VGG/Inception/ResNet/DenseNet
Stars: ✭ 314 (-75.24%)
Mutual labels: classification, resnet, densenet
Pytorch classification
利用pytorch实现图像分类的一个完整的代码，训练，预测，TTA，模型融合，模型部署，cnn提取特征，svm或者随机森林等进行分类，模型蒸馏，一个完整的代码
Stars: ✭ 395 (-68.85%)
Mutual labels: resnet, densenet, resnext
Mmclassification
OpenMMLab Image Classification Toolbox and Benchmark
Stars: ✭ 532 (-58.04%)
Mutual labels: resnet, imagenet, resnext
Senet Tensorflow
Simple Tensorflow implementation of "Squeeze and Excitation Networks" using Cifar10 (ResNeXt, Inception-v4, Inception-resnet-v2)
Stars: ✭ 682 (-46.21%)
Mutual labels: densenet, resnext
pytorch-classification
Classification on CIFAR-10/100 and ImageNet with PyTorch.
Features
- Unified interface for different network architectures
- Multi-GPU support
- Training progress bar with rich info
- Training log and training curve visualization code (see
./utils/logger.py)
Install
- Install PyTorch
- Clone recursively
git clone --recursive https://github.com/bearpaw/pytorch-classification.git
Training
Please see the Training recipes for how to train the models.
Results
CIFAR
Top1 error rate on the CIFAR-10/100 benchmarks are reported. You may get different results when training your models with different random seed. Note that the number of parameters are computed on the CIFAR-10 dataset.
| Model | Params (M) | CIFAR-10 (%) | CIFAR-100 (%) |
|---|---|---|---|
| alexnet | 2.47 | 22.78 | 56.13 |
| vgg19_bn | 20.04 | 6.66 | 28.05 |
| ResNet-110 | 1.70 | 6.11 | 28.86 |
| PreResNet-110 | 1.70 | 4.94 | 23.65 |
| WRN-28-10 (drop 0.3) | 36.48 | 3.79 | 18.14 |
| ResNeXt-29, 8x64 | 34.43 | 3.69 | 17.38 |
| ResNeXt-29, 16x64 | 68.16 | 3.53 | 17.30 |
| DenseNet-BC (L=100, k=12) | 0.77 | 4.54 | 22.88 |
| DenseNet-BC (L=190, k=40) | 25.62 | 3.32 | 17.17 |
ImageNet
Single-crop (224x224) validation error rate is reported.
| Model | Params (M) | Top-1 Error (%) | Top-5 Error (%) |
|---|---|---|---|
| ResNet-18 | 11.69 | 30.09 | 10.78 |
| ResNeXt-50 (32x4d) | 25.03 | 22.6 | 6.29 |
Pretrained models
Our trained models and training logs are downloadable at OneDrive.
Supported Architectures
CIFAR-10 / CIFAR-100
Since the size of images in CIFAR dataset is 32x32, popular network structures for ImageNet need some modifications to adapt this input size. The modified models is in the package models.cifar:
- [x] AlexNet
- [x] VGG (Imported from pytorch-cifar)
- [x] ResNet
- [x] Pre-act-ResNet
- [x] ResNeXt (Imported from ResNeXt.pytorch)
- [x] Wide Residual Networks (Imported from WideResNet-pytorch)
- [x] DenseNet
ImageNet
- [x] All models in
torchvision.models(alexnet, vgg, resnet, densenet, inception_v3, squeezenet) - [x] ResNeXt
- [ ] Wide Residual Networks
Contribute
Feel free to create a pull request if you find any bugs or you want to contribute (e.g., more datasets and more network structures).
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].