ImageClassification_Benchmark
Image classification workflow based on Pytorch 1.4 and Tensorflow 2.
Dataset
MNIST, CIFAR10, CIFAR100 and ImageNet are supported.
Features
Tensorboard, distributed training, mixed precision training, quantization and more feature are coming soon
Log
- 24 Mar Update:
- Update training/evaluation code
Todo list
- Pytorch
- Dataset ,dataloader, preprocessing and visualization
- Classical & state-of-the-art models(AlexNet, VGGNet, Inceptions, ResNet, ResNeXt, DenseNet, SeNet, MobileNet, e.t.c)
- Training and Evaluation codes
- Tensorboard support
- Binary quantization models
- Distributed training
- Arbitrary bits quantization
- Tensorflow
- Dataset ,dataloader, preprocessing and visualization
- Classical & state-of-the-art models(AlexNet, VGGNet, Inceptions, ResNet, ResNeXt, DenseNet, SeNet, MobileNet, e.t.c)
- Training and Evaluation codes
- Tensorboard support
- Binary quantization models
- Distributed training
- Arbitrary bits quantization
- Summary
- Mannual
- Result benchmarks