From 6e05e7452de8fc952043dfc33502509ca11a1c3e Mon Sep 17 00:00:00 2001 From: Setu Shah Date: Mon, 24 Jul 2023 10:19:41 -0700 Subject: [PATCH] fix: Use resolved `ultralytics` import --- dataquality/integrations/ultralytics.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/dataquality/integrations/ultralytics.py b/dataquality/integrations/ultralytics.py index f479cf74e..e1b3dc303 100644 --- 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]