From f3d381901ac87f2907522152602b0221b255b37a Mon Sep 17 00:00:00 2001
From: wanjunhui1 <wanjunhui1@sensetime.com>
Date: Tue, 3 Sep 2024 15:32:38 +0800
Subject: [PATCH] fix:fixed interruptions when there is no mask result for the
 current frame

---
 utils/common_utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/utils/common_utils.py b/utils/common_utils.py
index 9373bdc..ac60224 100644
--- a/utils/common_utils.py
+++ b/utils/common_utils.py
@@ -48,6 +48,10 @@ class CommonUtils:
                     object_mask = (mask == uid)
                     all_object_masks.append(object_mask[None])
             
+            if len(all_object_masks) == 0:
+                output_image_path = os.path.join(output_path, raw_image_name)
+                cv2.imwrite(output_image_path, image)
+                continue
             # get n masks: (n, h, w)
             all_object_masks = np.concatenate(all_object_masks, axis=0)