mediapipe_model_maker.object_detector.SupportedModels

Predefined object detector model specs supported by Model Maker.

Supported models include the following:

  • MOBILENET_V2: MobileNetV2 256x256 input
  • MOBILENET_V2_I320: MobileNetV2 320x320 input
  • MOBILENET_MULTI_AVG: MobileNet-MultiHW-AVG 256x256 input
  • MOBILENET_MULTI_AVG_I384: MobileNet-MultiHW-AVG 384x384 input

MOBILENET_MULTI_AVG <SupportedModels.MOBILENET_MULTI_AVG: functools.partial(<class 'mediapipe_model_maker.python.vision.object_detector.model_spec.ModelSpec'>, downloaded_files=DownloadedFiles(path='object_detector/mobilenetmultiavg', url='https://storage.googleapis.com/tf_model_garden/vision/qat/mobilenetv3.5_ssd_coco/mobilenetv3.5_ssd_i256_ckpt.tar.gz', is_folder=True), checkpoint_name='ckpt-277200', input_image_shape=[256, 256, 3], model_id='MobileNetMultiAVG', min_level=3, max_level=7)>
MOBILENET_MULTI_AVG_I384 <SupportedModels.MOBILENET_MULTI_AVG_I384: functools.partial(<class 'mediapipe_model_maker.python.vision.object_detector.model_spec.ModelSpec'>, downloaded_files=DownloadedFiles(path='object_detector/mobilenetmultiavg_i384', url='https://storage.googleapis.com/tf_model_garden/vision/qat/mobilenetv2_ssd_coco/mobilenetv3.5_ssd_i384_ckpt.tar.gz', is_folder=True), checkpoint_name='ckpt-277200', input_image_shape=[384, 384, 3], model_id='MobileNetMultiAVG', min_level=3, max_level=7)>
MOBILENET_V2 <SupportedModels.MOBILENET_V2: functools.partial(<class 'mediapipe_model_maker.python.vision.object_detector.model_spec.ModelSpec'>, downloaded_files=DownloadedFiles(path='object_detector/mobilenetv2_i256', url='https://storage.googleapis.com/tf_model_garden/vision/qat/mobilenetv2_ssd_coco/mobilenetv2_ssd_i256_ckpt.tar.gz', is_folder=True), checkpoint_name='ckpt-277200', input_image_shape=[256, 256, 3], model_id='MobileNetV2', min_level=3, max_level=7)>
MOBILENET_V2_I320 <SupportedModels.MOBILENET_V2_I320: functools.partial(<class 'mediapipe_model_maker.python.vision.object_detector.model_spec.ModelSpec'>, downloaded_files=DownloadedFiles(path='object_detector/mobilenetv2_i320', url='https://storage.googleapis.com/tf_model_garden/vision/qat/mobilenetv2_ssd_coco/mobilenetv2_ssd_i320_ckpt.tar.gz', is_folder=True), checkpoint_name='ckpt-277200', input_image_shape=[320, 320, 3], model_id='MobileNetV2', min_level=3, max_level=6)>