Google Colab

Requirements

pip install -r requirements.txt

Build model

MobileNetV3 Small

frommobilenetv3_factoryimportbuild_mobilenetv3model=build_mobilenetv3("small",input_shape=(224,224,3),num_classes=1001,width_multiplier=1.0,)

MobileNetV3 Large

frommobilenetv3_factoryimportbuild_mobilenetv3model=build_mobilenetv3("large",input_shape=(224,224,3),num_classes=1001,width_multiplier=1.0,)

Train

CIFAR10 dataset

python train.py \    --model_type small \    --width_multiplier 1.0 \    --height 128 \    --width 128 \    --dataset cifar10 \    --lr 0.01 \    --optimizer rmsprop \    --train_batch_size 256 \    --valid_batch_size 256 \    --num_epoch 10 \    --logdir logdir

MNIST dataset

python train.py \    --model_type small \    --width_multiplier 1.0 \    --height 128 \    --width 128 \    --dataset mnist \    --lr 0.01 \    --optimizer rmsprop \    --train_batch_size 256 \    --valid_batch_size 256 \    --num_epoch 10 \    --logdir logdir

Evaluate

CIFAR10 dataset

python evaluate.py \    --model_type small \    --width_multiplier 1.0 \    --height 128 \    --width 128 \    --dataset cifar10 \    --valid_batch_size 256 \    --model_path mobilenetv3_small_cifar10_10.h5

MNIST dataset

python evaluate.py \    --model_type small \    --width_multiplier 1.0 \    --height 128 \    --width 128 \    --dataset mnist \    --valid_batch_size 256 \    --model_path mobilenetv3_small_mnist_10.h5

TensorBoard

tensorboard --logdir logdir

License