Return metrics, Update docs (#846)

2023-02-08 05:33:25 +08:00
parent 3633d4c06b
commit 2e7a533ac3
9 changed files with 161 additions and 179 deletions
--- a/ultralytics/yolo/engine/model.py
+++ b/ultralytics/yolo/engine/model.py
@ -168,6 +168,7 @@ class YOLO:

        validator = self.ValidatorClass(args=args)
        validator(model=self.model)
+        return validator.metrics

    @smart_inference_mode()
    def export(self, **kwargs):