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

Question

Я запускаю 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)

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

Спасибо

Answer 1

Решить,

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)