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

Merge pull request #42 from Greywan/dev

Browse Source 
fix:fixed interruptions when there is no mask result for the current …
This commit is contained in:
Ren Tianhe
2024-09-03 16:07:00 +08:00
committed by GitHub
parent be1fa537bb f3d381901a
commit 5ee7526b79
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)