fix: Use resolved ultralytics import

rungalileo · Jul 24, 2023 · 6e05e74 · 6e05e74
1 parent a0a504f
commit 6e05e74
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/dataquality/integrations/ultralytics.py b/dataquality/integrations/ultralytics.py
@@ -3,9 +3,8 @@
 from urllib.parse import urljoin
 
 import torch
-import ultralytics
 from torchvision.ops.boxes import box_convert
-from ultralytics import YOLO
+from ultralytics import YOLO, checks
 from ultralytics.yolo.engine.predictor import BasePredictor
 from ultralytics.yolo.engine.trainer import BaseTrainer
 from ultralytics.yolo.engine.validator import BaseValidator
@@ -24,7 +23,7 @@
 from dataquality.utils.dqyolo import CONF_DEFAULT, IOU_DEFAULT
 from dataquality.utils.ultralytics import non_max_suppression, process_batch_data
 
-ultralytics.checks()
+checks()
 
 Coordinates = Union[Tuple, List]