Unofficial implementation of MobileNetV3 architecture described in paper Searching for MobileNetV3. This repository contains small and large MobileNetV3 architecture implemented using TensforFlow with tf.keras API.
Google Colab
MNIST
CIFAR10
Requirements
Python 3.6+
TensorFlow 1.13+
pip install -r requirements.txt
Build model
MobileNetV3 Small
from mobilenetv3_factory import build_mobilenetv3
model = build_mobilenetv3( "small", input_shape=(224, 224, 3), num_classes=1001, width_multiplier=1.0,
)
MobileNetV3 Large
from mobilenetv3_factory import build_mobilenetv3
model = build_mobilenetv3( "large", input_shape=(224, 224, 3), num_classes=1001, width_multiplier=1.0,
)
Graph, training and evaluaion metrics are saved to TensorBoard event file uder directory specified with --logdir` argument during training. You can launch TensorBoard using following command.