Updated some files
This commit is contained in:
parent
3fbb2b002b
commit
500ad5a7f0
|
@ -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",
|
||||
],
|
||||
)
|
||||
|
|
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue
Block a user