Make clear if mlmodel has NMS (#917)

apple · Jul 27, 2018 · dd9cebf · dd9cebf
1 parent 0588dca
commit dd9cebf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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'