Skip to content
Real-time object-detection using SSD on Mobilenet on iOS using CoreML, exported using tf-coreml
Branch: master
Clone or download
Latest commit 3f8858d Jan 9, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
SSDMobileNet Base project Jan 24, 2018
.gitignore
CODE_OF_CONDUCT.md Create CODE_OF_CONDUCT.md Jan 24, 2018
LICENSE added license-file Jan 9, 2019
README.md Update README.md Jan 26, 2018

README.md

SSDMobileNet_CoreML

Real-time object-detection on iOS using CoreML model of SSD based on Mobilenet. This project contains an example-project for running real-time inference of that model on iOS.

To add the preprocessing to the coreml-model, I included some additional parameters in the convert-step (image_scale and bias). To make use of CoreVision's abilities to convert image-format / scaling, I added the image_input_names parameter:

coreml_model = tfcoreml.convert(
      tf_model_path=frozen_model_file,
      mlmodel_path=coreml_model_file,
      input_name_shape_dict=input_tensor_shapes,
      image_input_names="Preprocessor/sub:0",
      output_feature_names=output_tensor_names,
      image_scale=2./255.,
      red_bias=-1.0,
      green_bias=-1.0,
      blue_bias=-1.0
)
You can’t perform that action at this time.