IDEA-Research / Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything
https://arxiv.org/abs/2401.14159
Apache License 2.0
15.16k stars 1.4k forks source link

为什么output中没有语义分割的数据 只有bbox的范围??求解答,我更改了multimask_output = TRUE,但是报错 #195

Open bigseamountain opened 1 year ago

bigseamountain commented 1 year ago

[{"value": 0, "label": "background"}, {"value": 1, "label": "bear", "logit": 0.63, "box": [658.04541015625, 876.801513671875, 1180.425048828125, 1230.06640625]}, {"value": 2, "label": "bear", "logit": 0.48, "box": [1116.91455078125, 444.0189208984375, 1666.591796875, 1204.355712890625]}, {"value": 3, "label": "bear", "logit": 0.57, "box": [892.4913940429688, 672.7112426757812, 1154.752197265625, 1014.0479736328125]}, {"value": 4, "label": "bear", "logit": 0.61, "box": [1198.3929443359375, 781.3394775390625, 1669.7503662109375, 1207.33740234375]}, {"value": 5, "label": "bear", "logit": 0.47, "box": [767.0352172851562, 739.9182739257812, 946.4960327148438, 897.655517578125]}]

bigseamountain commented 1 year ago

如果修改,修改那一部分呢? File "D:\Grounded-Segment-Anything-main\grounded_sam_demo.py", line 210, in show_mask(mask.cpu().numpy(), plt.gca(), random_color=True) File "D:\Grounded-Segment-Anything-main\grounded_sam_demo.py", line 92, in show_mask mask_image = mask.reshape(h, w, 1) * color.reshape(1, 1, -1) ValueError: cannot reshape array of size 8995800 into shape (1450,2068,1)