Zhen Ye commited on
Commit
3954f64
·
1 Parent(s): c7c9a25

fixed numpy error

Browse files
Files changed (1) hide show
  1. 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
- labels = [labels[0] for _ in range(len(result.masks or []))]
 
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