From d3ca3e4b9bd692a880f8ec6510d280d602fa10c6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 17 Jun 2023 19:35:19 +0200 Subject: [PATCH] Fix Pose val plotting bug (#3232) --- ultralytics/yolo/v8/pose/val.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/yolo/v8/pose/val.py b/ultralytics/yolo/v8/pose/val.py index 16afba4..f3fc1ac 100644 --- a/ultralytics/yolo/v8/pose/val.py +++ b/ultralytics/yolo/v8/pose/val.py @@ -154,7 +154,7 @@ class PoseValidator(DetectionValidator): def plot_predictions(self, batch, preds, ni): """Plots predictions for YOLO model.""" - pred_kpts = torch.cat([p[:, 6:].view(-1, *self.kpt_shape)[:15] for p in preds], 0) + pred_kpts = torch.cat([p[:, 6:].view(-1, *self.kpt_shape) for p in preds], 0) plot_images(batch['img'], *output_to_target(preds, max_det=self.args.max_det), kpts=pred_kpts,