Skip to content

Commit

Permalink
Fix visualizer in demo
Browse files Browse the repository at this point in the history
  • Loading branch information
sovrasov committed Jul 13, 2023
1 parent 280a276 commit 4d9d9ee
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ def render_result(self, results: Tuple[Any, dict]) -> np.ndarray:
predictions, frame_meta = results
if isinstance(self.converter, DetectionToAnnotationConverter):
# Predictions for the detection task
predictions = np.array([[pred.id, pred.score, *pred.get_coords()] for pred in predictions])
predictions = np.array([[pred.id, pred.score, *[pred.xmin, pred.ymin, pred.xmax, pred.ymax]] for pred in predictions.objects])
predictions.shape = len(predictions), 6
annotation_scene = self.converter.convert_to_annotation(predictions, frame_meta)
current_frame = frame_meta["frame"]
Expand Down

0 comments on commit 4d9d9ee

Please sign in to comment.