Update val max_dets=args.max_det=300 (#3051)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-07 08:17:39 +02:00 · 2023-06-07 08:17:39 +02:00 · 3ae81ee9d1
commit 3ae81ee9d1
parent 8ac8ff72ae
3 changed files with 10 additions and 9 deletions
--- a/ultralytics/yolo/v8/detect/val.py
+++ b/ultralytics/yolo/v8/detect/val.py
@ -223,7 +223,7 @@ class DetectionValidator(BaseValidator):
    def plot_predictions(self, batch, preds, ni):
        """Plots predicted bounding boxes on input images and saves the result."""
        plot_images(batch['img'],
-                    *output_to_target(preds, max_det=15),
+                    *output_to_target(preds, max_det=self.args.max_det),
                    paths=batch['im_file'],
                    fname=self.save_dir / f'val_batch{ni}_pred.jpg',
                    names=self.names,