From 500ad5a7f0263232bfa864968befce3ced6be010 Mon Sep 17 00:00:00 2001 From: kinaryml Date: Wed, 21 Sep 2022 03:43:18 -0700 Subject: [PATCH] Updated some files --- mediapipe/python/BUILD | 1 - mediapipe/tasks/python/test/vision/image_segmenter_test.py | 5 +++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mediapipe/python/BUILD b/mediapipe/python/BUILD index 331ee836..55c0a22e 100644 --- a/mediapipe/python/BUILD +++ b/mediapipe/python/BUILD @@ -86,7 +86,6 @@ cc_library( name = "builtin_task_graphs", deps = [ "//mediapipe/tasks/cc/vision/object_detector:object_detector_graph", - "//mediapipe/tasks/cc/vision/image_classification:image_classifier_graph", "//mediapipe/tasks/cc/vision/image_segmenter:image_segmenter_graph", ], ) diff --git a/mediapipe/tasks/python/test/vision/image_segmenter_test.py b/mediapipe/tasks/python/test/vision/image_segmenter_test.py index 704b0fc5..6194f766 100644 --- a/mediapipe/tasks/python/test/vision/image_segmenter_test.py +++ b/mediapipe/tasks/python/test/vision/image_segmenter_test.py @@ -104,10 +104,11 @@ class ImageSegmenterTest(parameterized.TestCase): segmenter = _ImageSegmenter.create_from_options(options) # Performs image segmentation on the input. - image_result = segmenter.segment(self.test_image) + category_masks = segmenter.segment(self.test_image) # Comparing results. - print(image_result) + print(len(category_masks)) + s # Closes the segmenter explicitly when the segmenter is not used in # a context.