xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-24 14:44:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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
Show Stats 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,