feat: logging ultralytics model path

This commit is contained in:
Dr.Lt.Data
2024-12-16 00:29:26 +09:00
parent b782e253ca
commit 3d94a74652
3 changed files with 6 additions and 2 deletions
+1 -1
View File
@@ -7,7 +7,7 @@
import importlib
version_code = [1, 0]
version_code = [1, 0, 1]
version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
print(f"### Loading: ComfyUI-Impact-Subpack ({version_str})")
+4
View File
@@ -1,12 +1,16 @@
import logging
import os
import folder_paths
from . import subcore
from . import utils
import logging
model_path = folder_paths.models_dir
utils.add_folder_path_and_extensions("ultralytics_bbox", [os.path.join(model_path, "ultralytics", "bbox")], folder_paths.supported_pt_extensions)
utils.add_folder_path_and_extensions("ultralytics_segm", [os.path.join(model_path, "ultralytics", "segm")], folder_paths.supported_pt_extensions)
utils.add_folder_path_and_extensions("ultralytics", [os.path.join(model_path, "ultralytics")], folder_paths.supported_pt_extensions)
logging.info(f'[Inspire Pack] ultralytics_bbox: {os.path.join(model_path, "ultralytics", "bbox")}')
logging.info(f'[Inspire Pack] ultralytics_segm: {os.path.join(model_path, "ultralytics", "segm")}')
class UltralyticsDetectorProvider:
+1 -1
View File
@@ -1,7 +1,7 @@
[project]
name = "comfyui-impact-subpack"
description = "This node pack provides nodes that complement the Impact Pack, such as the UltralyticsDetectorProvider."
version = "1.0"
version = "1.0.1"
license = { file = "LICENSE.txt" }
dependencies = ["matplotlib", "ultralytics==8.3.40"]