Skip to content
heliaEDGE
Reference
HELIA

mobilenet

This module provides utility functions to generate MobileNet models.

Parameters are in helia_edge.models.mobilenet_params.

Functions

Name Description
build MobileNetV1 model from MobileNetV1Params
mobilenetv1_layer Modified MobileNetV1 layer

Machine-readable model

  • mobilenetv1_layerfunctionModified MobileNetV1 MLPerf Tiny model for VWW: https://github.com/SiliconLabs/platformmlmodels/blob/master/eembc/Persondetection/mobilenetv1eembc.py
  • buildfunctionBuild a MobileNetV1 model.
function

Modified MobileNetV1 MLPerf Tiny model for VWW: https://github.com/SiliconLabs/platformmlmodels/blob/master/eembc/Persondetection/mobilenetv1eembc.py

helia_edge/models/mobilenet.py:21

mobilenetv1_layer(x: keras.KerasTensor, params: MobileNetV1Params) -> keras.KerasTensor

Modified MobileNetV1 MLPerf Tiny model for VWW: https://github.com/SiliconLabs/platform_ml_models/blob/master/eembc/Person_detection/mobilenet_v1_eembc.py

Parameters of mobilenetv1_layer
NameTypeDefaultDescription
xkeras.KerasTensorRequiredModel input
paramsMobileNetV1ParamsRequiredModel parameters
Returns of mobilenetv1_layer
TypeDescription
keras.KerasTensorkeras.KerasTensor: Model output
function

build

Python

Build a MobileNetV1 model.

helia_edge/models/mobilenet.py:270

build(
params: MobileNetV1Params,
input_shape: tuple[int | None, ...],
*,
batch_size: int | None = None,
name: str | None = None,
) -> keras.Model

Build a MobileNetV1 model.

Parameters of build
NameTypeDefaultDescription
paramsMobileNetV1ParamsRequiredModel parameters.
input_shapetuple[int | None, ...]RequiredInput shape without the batch axis; None for a variable axis.
batch_sizeint | NoneNoneStatic batch size; None for a dynamic batch.
namestr | NoneNoneModel name; the family when None.
Returns of build
TypeDescription
keras.Modelkeras.Model: The model, named ``mobilenet`` unless ``name`` is given.