replace np.int to int as np.int is deprecated and throws an error

AlibabaResearch · Mar 12, 2024 · f00cc20 · f00cc20
1 parent 5066d48
commit f00cc20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/DocumentUnderstanding/VGT/object_detection/ditod/Wordnn_embedding.py b/DocumentUnderstanding/VGT/object_detection/ditod/Wordnn_embedding.py
@@ -77,7 +77,7 @@ def forward(self, img, batched_inputs, stride = 1):
                     per_id = per_input_ids[word_idx]
 
                     bbox = per_input_bbox[word_idx] / stride
-                    w_start, h_start, w_end, h_end = bbox.round().astype(np.int).tolist()
+                    w_start, h_start, w_end, h_end = bbox.round().astype(int).tolist()
 
                     if self.use_UNK_text:
                         chargrid_map[iter_b, h_start:h_end, w_start: w_end] = 100