HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix Pose val plotting bug (#3232)

Browse source
This commit is contained in:
Glenn Jocher 2023-06-17 19:35:19 +02:00 • committed by GitHub
parent 1e5702a5b5
commit d3ca3e4b9b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/v8/pose/val.py
View file

@ -154,7 +154,7 @@ class PoseValidator(DetectionValidator):
def plot_predictions(self, batch, preds, ni): def plot_predictions(self, batch, preds, ni):
"""Plots predictions for YOLO model.""" """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'], plot_images(batch['img'],
*output_to_target(preds, max_det=self.args.max_det), *output_to_target(preds, max_det=self.args.max_det),
kpts=pred_kpts, kpts=pred_kpts,