ultralytics 8.0.67 Pose speeds, Comet and ClearML updates (#1871)

Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Victor Sonck <victor.sonck@gmail.com>
Co-authored-by: Danny Kim <dh031200@gmail.com>

tests/test_engine.py

@@ -2,7 +2,9 @@
 
 from pathlib import Path
 
+from ultralytics import YOLO
 from ultralytics.yolo.cfg import get_cfg
+from ultralytics.yolo.engine.exporter import Exporter
 from ultralytics.yolo.utils import DEFAULT_CFG, ROOT, SETTINGS
 from ultralytics.yolo.v8 import classify, detect, segment
 
@@ -14,20 +16,38 @@ MODEL = Path(SETTINGS['weights_dir']) / 'yolov8n'
 SOURCE = ROOT / 'assets'
 
 
+def test_func(model=None):
+    print('callback test passed')
+
+
+def test_export():
+    exporter = Exporter()
+    exporter.add_callback('on_export_start', test_func)
+    assert test_func in exporter.callbacks['on_export_start'], 'callback test failed'
+    f = exporter(model=YOLO(CFG_DET).model)
+    YOLO(f)(SOURCE)  # exported model inference
+
+
 def test_detect():
     overrides = {'data': 'coco8.yaml', 'model': CFG_DET, 'imgsz': 32, 'epochs': 1, 'save': False}
     CFG.data = 'coco8.yaml'
 
     # Trainer
     trainer = detect.DetectionTrainer(overrides=overrides)
+    trainer.add_callback('on_train_start', test_func)
+    assert test_func in trainer.callbacks['on_train_start'], 'callback test failed'
     trainer.train()
 
     # Validator
     val = detect.DetectionValidator(args=CFG)
+    val.add_callback('on_val_start', test_func)
+    assert test_func in val.callbacks['on_val_start'], 'callback test failed'
     val(model=trainer.best)  # validate best.pt
 
     # Predictor
     pred = detect.DetectionPredictor(overrides={'imgsz': [64, 64]})
+    pred.add_callback('on_predict_start', test_func)
+    assert test_func in pred.callbacks['on_predict_start'], 'callback test failed'
     result = pred(source=SOURCE, model=f'{MODEL}.pt')
     assert len(result), 'predictor test failed'
 
@@ -50,14 +70,20 @@ def test_segment():
 
     # trainer
     trainer = segment.SegmentationTrainer(overrides=overrides)
+    trainer.add_callback('on_train_start', test_func)
+    assert test_func in trainer.callbacks['on_train_start'], 'callback test failed'
     trainer.train()
 
     # Validator
     val = segment.SegmentationValidator(args=CFG)
+    val.add_callback('on_val_start', test_func)
+    assert test_func in val.callbacks['on_val_start'], 'callback test failed'
     val(model=trainer.best)  # validate best.pt
 
     # Predictor
     pred = segment.SegmentationPredictor(overrides={'imgsz': [64, 64]})
+    pred.add_callback('on_predict_start', test_func)
+    assert test_func in pred.callbacks['on_predict_start'], 'callback test failed'
     result = pred(source=SOURCE, model=f'{MODEL}-seg.pt')
     assert len(result), 'predictor test failed'
 
@@ -81,13 +107,19 @@ def test_classify():
 
     # Trainer
     trainer = classify.ClassificationTrainer(overrides=overrides)
+    trainer.add_callback('on_train_start', test_func)
+    assert test_func in trainer.callbacks['on_train_start'], 'callback test failed'
     trainer.train()
 
     # Validator
     val = classify.ClassificationValidator(args=CFG)
+    val.add_callback('on_val_start', test_func)
+    assert test_func in val.callbacks['on_val_start'], 'callback test failed'
     val(model=trainer.best)
 
     # Predictor
     pred = classify.ClassificationPredictor(overrides={'imgsz': [64, 64]})
+    pred.add_callback('on_predict_start', test_func)
+    assert test_func in pred.callbacks['on_predict_start'], 'callback test failed'
     result = pred(source=SOURCE, model=trainer.best)
     assert len(result), 'predictor test failed'