From 9c9519eeb81728de0fd1e528457768af4a0171a5 Mon Sep 17 00:00:00 2001
From: kinaryml <kinar.ravishankar@ymedialabs.com>
Date: Tue, 8 Nov 2022 02:55:19 -0800
Subject: [PATCH] Fixed a to_pb2() bug in the landmark_detection_result
 container

---
 .../containers/landmark_detection_result.py   | 28 +++++++++++++------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/mediapipe/tasks/python/components/containers/landmark_detection_result.py b/mediapipe/tasks/python/components/containers/landmark_detection_result.py
index e058f84a8..317290811 100644
--- a/mediapipe/tasks/python/components/containers/landmark_detection_result.py
+++ b/mediapipe/tasks/python/components/containers/landmark_detection_result.py
@@ -53,19 +53,29 @@ class LandmarksDetectionResult:
   def to_pb2(self) -> _LandmarksDetectionResultProto:
     """Generates a LandmarksDetectionResult protobuf object."""
 
+    landmarks = _NormalizedLandmarkListProto()
+    landmarks.landmark.extend([
+      landmark.to_pb2() for landmark in self.landmarks
+    ])
+
     classifications = _ClassificationListProto()
-    for category in self.categories:
-      classifications.classification.append(
-          _ClassificationProto(
-              index=category.index,
-              score=category.score,
-              label=category.category_name,
-              display_name=category.display_name))
+    classifications.classification.extend([
+      _ClassificationProto(
+        index=category.index,
+        score=category.score,
+        label=category.category_name,
+        display_name=category.display_name) for category in self.categories
+    ])
+
+    world_landmarks = _LandmarkListProto()
+    world_landmarks.landmark.extend([
+      world_landmark.to_pb2() for world_landmark in self.world_landmarks
+    ])
 
     return _LandmarksDetectionResultProto(
-        landmarks=_NormalizedLandmarkListProto(self.landmarks),
+        landmarks=landmarks,
         classifications=classifications,
-        world_landmarks=_LandmarkListProto(self.world_landmarks),
+        world_landmarks=world_landmarks,
         rect=self.rect.to_pb2())
 
   @classmethod