From 6306aab1e4984c56b3ccdbb4bd191b3578aad7af Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Mon, 4 Nov 2024 08:04:25 -0500 Subject: [PATCH] detailer add ultralytics version Signed-off-by: Vladimir Mandic --- modules/postprocess/yolo.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/postprocess/yolo.py b/modules/postprocess/yolo.py index 14d6558be..1a52e1334 100644 --- a/modules/postprocess/yolo.py +++ b/modules/postprocess/yolo.py @@ -156,10 +156,10 @@ class YoloRestorer(Detailer): try: model_file = modelloader.load_file_from_url(url=model_url, model_dir=shared.opts.yolo_dir, file_name=file_name) if model_file is not None: - from ultralytics import YOLO # pylint: disable=import-outside-toplevel - model = YOLO(model_file) + import ultralytics + model = ultralytics.YOLO(model_file) classes = list(model.names.values()) - shared.log.info(f'Load: type=Detailer name="{model_name}" model="{model_file}" classes={classes}') + shared.log.info(f'Load: type=Detailer name="{model_name}" model="{model_file}" ultralytics={ultralytics.__version__} classes={classes}') self.models[model_name] = model return model_name, model except Exception as e: