imgsz warning fix, download function consolidation (#681)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: HaeJin Lee <seareale@gmail.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>

ultralytics/yolo/v8/classify/predict.py

@@ -74,7 +74,7 @@ def predict(cfg=DEFAULT_CFG, use_python=False):
         from ultralytics import YOLO
         YOLO(model)(**args)
     else:
-        predictor = ClassificationPredictor(args)
+        predictor = ClassificationPredictor(overrides=args)
         predictor.predict_cli()
 
 

ultralytics/yolo/v8/classify/train.py

@@ -146,7 +146,7 @@ def train(cfg=DEFAULT_CFG, use_python=False):
         from ultralytics import YOLO
         YOLO(model).train(**args)
     else:
-        trainer = ClassificationTrainer(args)
+        trainer = ClassificationTrainer(overrides=args)
         trainer.train()
 
 

ultralytics/yolo/v8/detect/predict.py

@@ -92,7 +92,7 @@ def predict(cfg=DEFAULT_CFG, use_python=False):
         from ultralytics import YOLO
         YOLO(model)(**args)
     else:
-        predictor = DetectionPredictor(args)
+        predictor = DetectionPredictor(overrides=args)
         predictor.predict_cli()
 
 

ultralytics/yolo/v8/detect/train.py

@@ -204,7 +204,7 @@ def train(cfg=DEFAULT_CFG, use_python=False):
         from ultralytics import YOLO
         YOLO(model).train(**args)
     else:
-        trainer = DetectionTrainer(args)
+        trainer = DetectionTrainer(overrides=args)
         trainer.train()
 
 

ultralytics/yolo/v8/segment/predict.py

@@ -110,7 +110,7 @@ def predict(cfg=DEFAULT_CFG, use_python=False):
         from ultralytics import YOLO
         YOLO(model)(**args)
     else:
-        predictor = SegmentationPredictor(args)
+        predictor = SegmentationPredictor(overrides=args)
         predictor.predict_cli()
 
 

ultralytics/yolo/v8/segment/train.py

@@ -150,7 +150,7 @@ def train(cfg=DEFAULT_CFG, use_python=False):
         from ultralytics import YOLO
         YOLO(model).train(**args)
     else:
-        trainer = SegmentationTrainer(args)
+        trainer = SegmentationTrainer(overrides=args)
         trainer.train()