Removed an unnecessary check and updated tests to check if the masks are generated or not

2023-04-24 11:23:27 -07:00
parent 9032bce577
commit b511822815
2 changed files with 55 additions and 19 deletions
@@ -263,7 +263,6 @@ class PoseLandmarker(base_vision_task_api.BaseVisionTaskApi):
      )

    output_streams = [
-        ':'.join([_SEGMENTATION_MASK_TAG, _SEGMENTATION_MASK_STREAM_NAME]),
        ':'.join([_NORM_LANDMARKS_TAG, _NORM_LANDMARKS_STREAM_NAME]),
        ':'.join([
          _POSE_WORLD_LANDMARKS_TAG, _POSE_WORLD_LANDMARKS_STREAM_NAME