Project import generated by Copybara.

GitOrigin-RevId: 1610e588e497817fae2d9a458093ab6a370e2972

mediapipe/modules/face_landmark/BUILD

@@ -101,6 +101,7 @@ mediapipe_simple_subgraph(
     register_as = "FaceLandmarkFrontCpuImage",
     deps = [
         ":face_landmark_front_cpu",
+        "//mediapipe/calculators/image:image_transformation_calculator",
         "//mediapipe/calculators/util:from_image_calculator",
     ],
 )
@@ -111,6 +112,7 @@ mediapipe_simple_subgraph(
     register_as = "FaceLandmarkFrontGpuImage",
     deps = [
         ":face_landmark_front_gpu",
+        "//mediapipe/calculators/image:image_transformation_calculator",
         "//mediapipe/calculators/util:from_image_calculator",
     ],
 )

mediapipe/modules/face_landmark/face_landmark_front_cpu_image.pbtxt

@@ -8,6 +8,8 @@ input_stream: "IMAGE:image"
 # Max number of faces to detect/track. (int)
 input_side_packet: "NUM_FACES:num_faces"
 
+# The original input image. (Image)
+output_stream: "IMAGE:image"
 # Collection of detected/predicted faces, each represented as a list of 468 face
 # landmarks. (std::vector<NormalizedLandmarkList>)
 # NOTE: there will not be an output packet in the LANDMARKS stream for this
@@ -30,7 +32,20 @@ output_stream: "ROIS_FROM_DETECTIONS:face_rects_from_detections"
 node {
   calculator: "FromImageCalculator"
   input_stream: "IMAGE:image"
-  output_stream: "IMAGE_CPU:image_frame"
+  output_stream: "IMAGE_CPU:raw_image_frame"
+  output_stream: "SOURCE_ON_GPU:is_gpu_image"
+}
+
+# TODO: Remove the extra flipping once adopting MlImage.
+# If the source images are on gpu, flip the data vertically before sending them
+# into FaceLandmarkFrontCpu. This maybe needed because OpenGL represents images
+# assuming the image origin is at the bottom-left corner, whereas MediaPipe in
+# general assumes the image origin is at the top-left corner.
+node: {
+  calculator: "ImageTransformationCalculator"
+  input_stream: "IMAGE:raw_image_frame"
+  input_stream: "FLIP_VERTICALLY:is_gpu_image"
+  output_stream: "IMAGE:image_frame"
 }
 
 node {

mediapipe/modules/face_landmark/face_landmark_front_gpu_image.pbtxt

@@ -8,6 +8,8 @@ input_stream: "IMAGE:image"
 # Max number of faces to detect/track. (int)
 input_side_packet: "NUM_FACES:num_faces"
 
+# The original input image. (Image)
+output_stream: "IMAGE:image"
 # Collection of detected/predicted faces, each represented as a list of 468 face
 # landmarks. (std::vector<NormalizedLandmarkList>)
 # NOTE: there will not be an output packet in the LANDMARKS stream for this
@@ -30,6 +32,19 @@ output_stream: "ROIS_FROM_DETECTIONS:face_rects_from_detections"
 node {
   calculator: "FromImageCalculator"
   input_stream: "IMAGE:image"
+  output_stream: "IMAGE_GPU:raw_gpu_buffer"
+  output_stream: "SOURCE_ON_GPU:is_gpu_image"
+}
+
+# TODO: Remove the extra flipping once adopting MlImage.
+# If the source images are on gpu, flip the data vertically before sending them
+# into FaceLandmarkFrontGpu. This maybe needed because OpenGL represents images
+# assuming the image origin is at the bottom-left corner, whereas MediaPipe in
+# general assumes the image origin is at the top-left corner.
+node: {
+  calculator: "ImageTransformationCalculator"
+  input_stream: "IMAGE_GPU:raw_gpu_buffer"
+  input_stream: "FLIP_VERTICALLY:is_gpu_image"
   output_stream: "IMAGE_GPU:gpu_buffer"
 }