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,