talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add custom metadata for object detection model with out-of-graph nms.

Browse Source 
PiperOrigin-RevId: 527083453
This commit is contained in:
MediaPipe Team
2023-04-25 14:58:51 -07:00
committed by Copybara-Service
parent 17f5b95387
commit 507ed0d91d
15 changed files with 19798 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/model_maker/python/vision/object_detector/object_detector.py
+1 -1
View File
@@ -328,7 +328,7 @@ class ObjectDetector(classifier.Classifier):
converter.target_spec.supported_ops = (tf.lite.OpsSet.TFLITE_BUILTINS,)
tflite_model = converter.convert()
writer = object_detector_writer.MetadataWriter.create(
writer = object_detector_writer.MetadataWriter.create_for_models_with_nms(
tflite_model,
self._model_spec.mean_rgb,
self._model_spec.stddev_rgb,