From dd9cebf0d9731afaa9d3a9a14380fc331d2d8f12 Mon Sep 17 00:00:00 2001 From: Gustav Larsson Date: Fri, 27 Jul 2018 16:53:25 -0700 Subject: [PATCH] Make clear if mlmodel has NMS (#917) --- .../turicreate/toolkits/object_detector/object_detector.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/unity/python/turicreate/toolkits/object_detector/object_detector.py b/src/unity/python/turicreate/toolkits/object_detector/object_detector.py index 8af37bc645..5778e59e3c 100644 --- a/src/unity/python/turicreate/toolkits/object_detector/object_detector.py +++ b/src/unity/python/turicreate/toolkits/object_detector/object_detector.py @@ -1554,6 +1554,8 @@ def export_coreml(self, filename, confidenceThresholdString = ('(optional)' + ' Confidence Threshold override (default: {})') model_type = 'object detector (%s)' % self.model + if include_non_maximum_suppression: + model_type += ' with non-maximum suppression' model.description.metadata.shortDescription = \ _coreml_utils._mlmodel_short_description(model_type) model.description.input[0].shortDescription = 'Input image'