mobilenet(v1).py - academic (idiomatic)
mobilenet(v1)_c.py - production (composable)
mobilenet(v2).py - academic (idiomatic)
mobilenet(v2)_c.py - production (composable)
mobilenet(v3).py - academic (idiomatic)
mobilenet(v3)_c.py - production (composable)
Example: Instantiate a stock MobileNet V2 model
from mobilenet_v2_c import MobileNetV2
# MobileNet v2.0 from research paper
mobilenet = MobileNetV2()
# MobileNet v2.0 custom input shape/classes
mobilenet = MobileNetV2(input_shape=(128, 128, 3), n_classes=50)
# getter for the tf.keras model
model = mobilenet.model
Example: Composable and Train a MobileNet V1 model
''' Example for constructing/training a MobileNet V1 model on CIFAR-10
'''
# Example of constructing a mini-MobileNet
groups = [ { 'n_filters': 128, 'n_blocks': 1 },
{ 'n_filters': 256, 'n_blocks': 1 },
{ 'n_filters': 1024, 'n_blocks': 2 } ]
mobilenet = MobileNetV1(groups, input_shape=(32, 32, 3), n_classes=10)
mobilenet.model.summary()
mobilenet.cifar10()