Model spec architecture and the names to use

TensorFlow General Discussion
, ,
1

I am trying to use the Object Detection with TensorFlow Lite Model Maker colab and was wondering how to specify a different object detection architecture…spec = …
Can I specify a different architecture? The only ones that I have had success with is the EfficientDet family [0 - 4].

Does anyone know which models on the tensorflow hub can be used for object detection that the names that should be specified in the spec statement?

1 Like
3

@Winton_Cape,

Tensorflow Lite Model Maker only supports the following object detection models.

OBJECT_DETECTION_MODELS = [
    'efficientdet_lite0',
    'efficientdet_lite1',
    'efficientdet_lite2',
    'efficientdet_lite3',
    'efficientdet_lite4',
]

Thank you!