|
|
@ -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,
|
|
|
|