Skip to content

GitLab

Commit b028d543 authored by Louis Maddox's avatar Louis Maddox Committed by ericmintun
Browse files

Fix another instance of the same bug

parent 6325eb85
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
segment_anything/automatic_mask_generator.py
View file @ b028d543
...@@ -357,7 +357,7 @@ class SamAutomaticMaskGenerator: ...@@ -357,7 +357,7 @@ class SamAutomaticMaskGenerator:
keep_by_nms = batched_nms( keep_by_nms = batched_nms(
boxes.float(), boxes.float(),
torch.as_tensor(scores), torch.as_tensor(scores),
torch.zeros(len(boxes)), # categories torch.zeros_like(boxes[:,0]), # categories
iou_threshold=nms_thresh, iou_threshold=nms_thresh,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment