nodes/SuperMasterBlasterLaser-ComfyUI_YOLO_Classifiers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added predict

Browse Source
This commit is contained in:
DESKTOP-UL3K6JV\Daniq
2024-06-16 16:11:13 +05:00
parent c83cb1ba5b
commit 9b67673d83
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
classifiers.py
+1 -1
View File
@@ -54,7 +54,7 @@ class YOLOClassify:
CATEGORY = "SuperMasterBlasterLaser/ComfyUI_YOLO_Classifiers"
def classify(self, yolo_classifier_model, image_to_classify):
result = yolo_classifier_model(image_to_classify)
result = yolo_classifier_model.predict(image_to_classify)
class_name = None
for r in result: