config.json

{"model_type":"mobilenet_v2",
    "model_key": "mobilemodel",
    "description": "My MobileNet-V2 model trained on my dataset",
    "framework": "tensorflow",
    "model": {
        "path": "mobilemodel",
        "class": "TFMobileNetV2",
        "config": {
            "input_shape": [128, 128, 3],
            "weights": "imagenet",
            "classes": 2
        }
    },
    "preprocessing": {
        "name": "tf_mobilenet_v2",
        "input_shape": [128, 128, 3]
    }
}