uzh-rpg/RVT

Visualization problems

Opened this issue · 1 comments

EyoungZhang commented

Hello, could you please provide me with the visual code? Thanks

magehrig commented

Same issue as #9
If you are interested how visualization is done in the code:

RVT/callbacks/detection.py

Lines 76 to 80 in 5774d5c

ev_img = self.ev_repr_to_img(ev_tensor.cpu().numpy())
predictions_proph = outputs[ObjDetOutput.PRED_PROPH]
prediction_img = ev_img.copy()
draw_bboxes(prediction_img, predictions_proph, labelmap=self.label_map)