Mask pycocotools (#116)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-30 01:01:03 +08:00 · 2022-12-30 01:01:03 +08:00 · a9dc1637c2
commit a9dc1637c2
parent 37e0f6e0e4
2 changed files with 67 additions and 14 deletions
--- a/ultralytics/yolo/engine/trainer.py
+++ b/ultralytics/yolo/engine/trainer.py
@ -423,6 +423,7 @@ class BaseTrainer:
                strip_optimizer(f)  # strip optimizers
                if f is self.best:
                    self.console.info(f'\nValidating {f}...')
+                    self.validator.args.save_json = True
                    self.metrics = self.validator(model=f)
                    self.metrics.pop('fitness', None)
                    self.trigger_callbacks('on_val_end')