Updated some files

This commit is contained in:
kinaryml 2022-09-21 03:43:18 -07:00
parent 3fbb2b002b
commit 500ad5a7f0
2 changed files with 3 additions and 3 deletions

View File

@ -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",
],
)

View File

@ -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.