fix: dispose segmentation mask packets (#1066)

homuler · Dec 16, 2023 · 370bc5c · 370bc5c
1 parent 3b1f0ec
commit 370bc5c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/...om.github.homuler.mediapipe/Runtime/Scripts/Tasks/Vision/PoseLandmarker/PoseLandmarker.cs b/...om.github.homuler.mediapipe/Runtime/Scripts/Tasks/Vision/PoseLandmarker/PoseLandmarker.cs
@@ -200,7 +200,15 @@ private static PoseLandmarkerResult BuildPoseLandmarkerResult(PacketMap outputPa
 
       var poseLandmarksProto = poseLandmarksProtoPacket.Get();
       var poseWorldLandmarksProto = outputPackets.At<LandmarkListVectorPacket, List<LandmarkList>>(_POSE_WORLD_LANDMARKS_STREAM_NAME).Get();
-      var segmentationMasks = outputPackets.At<ImageVectorPacket, List<Image>>(_SEGMENTATION_MASK_STREAM_NAME)?.Get();
+
+      List<Image> segmentationMasks = null;
+      using (var segmentationMaskPacket = outputPackets.At<ImageVectorPacket, List<Image>>(_SEGMENTATION_MASK_STREAM_NAME))
+      {
+        if (segmentationMaskPacket != null)
+        {
+          segmentationMasks = segmentationMaskPacket.Get();
+        }
+      }
 
       return PoseLandmarkerResult.CreateFrom(poseLandmarksProto, poseWorldLandmarksProto, segmentationMasks);
     }