quoccuongle/Grounded-SAM-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix:fixed interruptions when there is no mask result for the current frame

Browse Source
This commit is contained in:
wanjunhui1
2024-09-03 15:32:38 +08:00
parent be1fa537bb
commit f3d381901a
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/common_utils.py
View File

@@ -48,6 +48,10 @@ class CommonUtils:
object_mask = (mask == uid) object_mask = (mask == uid)
all_object_masks.append(object_mask[None]) 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) # get n masks: (n, h, w)
all_object_masks = np.concatenate(all_object_masks, axis=0) all_object_masks = np.concatenate(all_object_masks, axis=0)