Визуализатор Detectron2 - объект 'Visualizer' не имеет атрибута 'get_image' - PullRequest
Новая
       81

Визуализатор Detectron2 - объект 'Visualizer' не имеет атрибута 'get_image'

0 голосов
/ 03 августа 2020

Я запускаю detectron2 для обнаружения объектов.

В оригинале после трининга я запускаю следующий код: 

v = Visualizer(im[:, :, ::-1],
            metadata=MetadataCatalog.get("name"), 
            scale=1, 
            instance_mode=ColorMode.SEGMENTATION)
v = v.draw_instance_predictions(outputs["instances"].to("cpu"))
cv2_imshow(v.get_image()[:, :, ::-1])
cv2_imshow(resized)

, и он отлично работает.

Теперь, после фильтрации некоторых сегментированных объектов, я пытаюсь запустить код и представить только некоторые из сегментов, поэтому я построил массив arr_in , который включает только количество экземпляров, которые я хочу представить. и я пытаюсь добавить его в v.draw_instance_prediction. 

v = Visualizer(im[:, :, ::-1],
            metadata=MetadataCatalog.get("name"), 
            scale=1, 
            instance_mode=ColorMode.SEGMENTATION   
     )
for i in range(len(arr_in)):
  num= np.int(arr_in[i])
  np.append(v, v.draw_instance_predictions(outputs["instances"].to("cpu")[num]))

cv2_imshow(v.get_image()[:, :, ::-1])
cv2_imshow(resized)

, но это не работает. Я спрашиваю, как мне добавлять переменные из предсказаний типов?

Спасибо

1 Ответ

0 голосов
/ 03 августа 2020

Решить, 

v = Visualizer(im[:, :, ::-1],
            metadata=MetadataCatalog.get("name"), 
            scale=1, 
            instance_mode=ColorMode.SEGMENTATION   
)
for i in range(len(arr_in)):
  num= np.int(arr_in[i])
  #print(num)
  np.append(v.draw_instance_predictions(outputs["instances"].to("cpu")), 
     v.draw_instance_predictions(outputs["instances"].to("cpu")[num]))

v = v.draw_instance_predictions(outputs["instances"].to("cpu"))
cv2_imshow(v.get_image()[:, :, ::-1])
cv2_imshow(resized)
...