Spaces:
Paused
Paused
Zhen Ye
commited on
Commit
·
3954f64
1
Parent(s):
c7c9a25
fixed numpy error
Browse files- inference.py +2 -1
inference.py
CHANGED
|
@@ -204,7 +204,8 @@ def infer_segmentation_frame(
|
|
| 204 |
result = segmenter.predict(frame, text_prompts=text_queries)
|
| 205 |
labels = text_queries or []
|
| 206 |
if len(labels) == 1:
|
| 207 |
-
|
|
|
|
| 208 |
return draw_masks(frame, result.masks, labels=labels), result
|
| 209 |
|
| 210 |
|
|
|
|
| 204 |
result = segmenter.predict(frame, text_prompts=text_queries)
|
| 205 |
labels = text_queries or []
|
| 206 |
if len(labels) == 1:
|
| 207 |
+
masks = result.masks if result.masks is not None else []
|
| 208 |
+
labels = [labels[0] for _ in range(len(masks))]
|
| 209 |
return draw_masks(frame, result.masks, labels=labels), result
|
| 210 |
|
| 211 |
|