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

Added Image Segmenter implementation and tests

Browse Source
This commit is contained in:
kinaryml
2022-09-25 09:16:13 -07:00
parent e028b24c42
commit d25626ff63
4 changed files with 92 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/python/vision/image_segmenter.py
+2 -2
View File
@@ -128,7 +128,7 @@ class ImageSegmenter(base_vision_task_api.BaseVisionTaskApi):
def packets_callback(output_packets: Mapping[str, packet_module.Packet]):
if output_packets[_IMAGE_OUT_STREAM_NAME].is_empty():
return
segmentation_result = packet_getter.get_proto_list(
segmentation_result = packet_getter.get_image_list(
output_packets[_SEGMENTATION_OUT_STREAM_NAME])
image = packet_getter.get_image(output_packets[_IMAGE_OUT_STREAM_NAME])
timestamp = output_packets[_IMAGE_OUT_STREAM_NAME].timestamp
@@ -166,6 +166,6 @@ class ImageSegmenter(base_vision_task_api.BaseVisionTaskApi):
"""
output_packets = self._process_image_data(
{_IMAGE_IN_STREAM_NAME: packet_creator.create_image(image)})
segmentation_result = packet_getter.get_proto_list(
segmentation_result = packet_getter.get_image_list(
output_packets[_SEGMENTATION_OUT_STREAM_NAME])
return segmentation_result