Skip to content

Commit e444834

Browse files
committed
Fix np warning
1 parent 3726152 commit e444834

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/python/model_api/tilers/instance_segmentation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def _merge_results(self, results, shape) -> InstanceSegmentationResult:
123123
labels = labels.astype(np.int32)
124124
resized_masks, label_names = [], []
125125
for mask, box, label_idx in zip(masks, bboxes, labels):
126-
label_names.append(self.model.labels[int(label_idx)])
126+
label_names.append(self.model.labels[int(label_idx.squeeze())])
127127
resized_masks.append(_segm_postprocess(box, mask, *shape[:-1]))
128128

129129
resized_masks = np.stack(resized_masks) if resized_masks else masks

0 commit comments

Comments
 (0)