Add benchmarks to Docker publish workflow (#3931)

2023-07-25 04:16:02 +02:00
parent 2ee147838a
commit 9f5ab67ba2
9 changed files with 54 additions and 43 deletions
--- a/ultralytics/engine/model.py
+++ b/ultralytics/engine/model.py
@ -319,7 +319,11 @@ class YOLO:
        overrides.update(kwargs)
        overrides['mode'] = 'benchmark'
        overrides = {**DEFAULT_CFG_DICT, **overrides}  # fill in missing overrides keys with defaults
-        return benchmark(model=self, imgsz=overrides['imgsz'], half=overrides['half'], device=overrides['device'])
+        return benchmark(model=self,
+                         imgsz=overrides['imgsz'],
+                         half=overrides['half'],
+                         device=overrides['device'],
+                         verbose=overrides['verbose'])

    def export(self, **kwargs):
        """