Fix for compatible with numpy

microsoft · Nov 7, 2024 · 4cb6326 · 4cb6326
1 parent b24a7c5
commit 4cb6326
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/modeling/architectures/seem_model_demo.py b/modeling/architectures/seem_model_demo.py
@@ -734,7 +734,7 @@ def prepare_targets(self, batched_inputs, images):
                 tokens = gtext['tokens']
 
                 unique_hash_id = np.unique(grd_hash, return_index=True)[1]
-                selected_mask = np.zeros(len(grd_hash)).astype(np.bool)
+                selected_mask = np.zeros(len(grd_hash)).astype(bool)
                 selected_mask[unique_hash_id] = True
 
                 selected_token_emb = token_emb[selected_mask]

diff --git a/modeling/architectures/seem_model_v1.py b/modeling/architectures/seem_model_v1.py
@@ -957,7 +957,7 @@ def prepare_targets(self, batched_inputs, images):
                 tokens = gtext['tokens']
 
                 unique_hash_id = np.unique(grd_hash, return_index=True)[1]
-                selected_mask = np.zeros(len(grd_hash)).astype(np.bool)
+                selected_mask = np.zeros(len(grd_hash)).astype(bool)
                 selected_mask[unique_hash_id] = True
 
                 selected_token_emb = token_emb[selected_mask]