Import DEFAULT_CONFIG from utils/__init__.py (#115)

2022-12-29 14:58:03 +01:00
parent 92dad1c1b5
commit 37e0f6e0e4
8 changed files with 8 additions and 13 deletions
--- a/ultralytics/yolo/v8/detect/predict.py
+++ b/ultralytics/yolo/v8/detect/predict.py
@ -2,8 +2,7 @@ import hydra
 import torch

 from ultralytics.yolo.engine.predictor import BasePredictor
-from ultralytics.yolo.engine.trainer import DEFAULT_CONFIG
-from ultralytics.yolo.utils import ops
+from ultralytics.yolo.utils import DEFAULT_CONFIG, ops
 from ultralytics.yolo.utils.plotting import Annotator, colors, save_one_box