Skip to content

Commit 87b112a

Browse files
committed
fix numpy error
1 parent adfa58d commit 87b112a

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tasks/interactive_idino_m2m.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def interactive_infer_image(model, image,all_classes,all_parts, thresh,text_size
8383
areas.append(area)
8484
mask,_=remove_small_regions(mask,int(hole_scale),mode="holes")
8585
mask,_=remove_small_regions(mask,int(island_scale),mode="islands")
86-
mask=(mask).astype(np.float)
86+
mask=(mask).astype(float)
8787
out_txt = texts
8888
visual = Visualizer(image_ori, metadata=metadata)
8989
color=[0.,0.,1.0]

tasks/interactive_predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def process_multi_mask(self, masks, ious, image_ori):
8585
areas.append(area)
8686
mask, _ = self.remove_small_regions(mask, int(self.hole_scale), mode="holes")
8787
mask, _ = self.remove_small_regions(mask, int(self.island_scale), mode="islands")
88-
mask = (mask).astype(np.float)
88+
mask = (mask).astype(float)
8989
out_txt = texts
9090
visual = Visualizer(image_ori, metadata=metadata)
9191
color = [0., 0., 1.0]

0 commit comments

Comments
 (0)