Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2074152852 | ||
|
|
979f356783 |
+1
-1
@@ -8,7 +8,7 @@
|
||||
import importlib
|
||||
import logging
|
||||
|
||||
version_code = [1, 2, 7]
|
||||
version_code = [1, 2, 9]
|
||||
version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
|
||||
logging.info(f"### Loading: ComfyUI-Impact-Subpack ({version_str})")
|
||||
|
||||
|
||||
@@ -30,6 +30,22 @@ class UltralyticsDetectorProvider:
|
||||
|
||||
def doit(self, model_name):
|
||||
model_path = folder_paths.get_full_path("ultralytics", model_name)
|
||||
|
||||
if model_path is None:
|
||||
print(f"[Impact Subpack] model file '{model_name}' is not found in one of the following directories:")
|
||||
|
||||
cands = []
|
||||
cands.extend(folder_paths.get_folder_paths("ultralytics"))
|
||||
if model_name.startswith('bbox/'):
|
||||
cands.extend(folder_paths.get_folder_paths("ultralytics_bbox"))
|
||||
elif model_name.startswith('segm/'):
|
||||
cands.extend(folder_paths.get_folder_paths("ultralytics_segm"))
|
||||
|
||||
formatted_cands = "\n\t".join(cands)
|
||||
print(f'\t{formatted_cands}\n')
|
||||
|
||||
raise ValueError(f"[Impact Subpack] model file '{model_name}' is not found.")
|
||||
|
||||
model = subcore.load_yolo(model_path)
|
||||
|
||||
if model_name.startswith("bbox"):
|
||||
|
||||
+1
-1
@@ -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.2.7"
|
||||
version = "1.2.9"
|
||||
license = { file = "LICENSE.txt" }
|
||||
dependencies = ["matplotlib", "ultralytics==8.3.40"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user