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

fix: zero object detection error (#64)

Browse Source 
* update dockerfile

* fix: zero object detection error

* fix: zero object detection error
This commit is contained in:
Susan Shen
2024-10-30 13:38:30 +01:00
committed by GitHub
parent e537a1e763
commit bf57b3086c
4 changed files with 190 additions and 152 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
utils/mask_dictionary_model.py
View File

@@ -84,6 +84,25 @@ class MaskDictionaryModel:
iou = intersection / union
return iou
def save_empty_mask_and_json(self, mask_data_dir, json_data_dir, image_name_list=None):
mask_img = torch.zeros((self.mask_height, self.mask_width))
if image_name_list:
for image_base_name in image_name_list:
image_base_name = image_base_name.split(".")[0]+".npy"
mask_name = "mask_"+image_base_name
np.save(os.path.join(mask_data_dir, mask_name), mask_img.numpy().astype(np.uint16))
json_data_path = os.path.join(json_data_dir, mask_name.replace(".npy", ".json"))
print("save_empty_mask_and_json", json_data_path)
self.to_json(json_data_path)
else:
np.save(os.path.join(mask_data_dir, self.mask_name), mask_img.numpy().astype(np.uint16))
json_data_path = os.path.join(json_data_dir, self.mask_name.replace(".npy", ".json"))
print("save_empty_mask_and_json", json_data_path)
self.to_json(json_data_path)
def to_dict(self):
return {
"mask_name": self.mask_name,