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

Changed permute

Browse Source
This commit is contained in:
DESKTOP-UL3K6JV\Daniq
2024-06-16 17:09:24 +05:00
parent 19c20a53d8
commit d9f85fcf1e
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
@@ -55,7 +55,7 @@ class YOLOClassify:
CATEGORY = "SuperMasterBlasterLaser/ComfyUI_YOLO_Classifiers"
def classify(self, yolo_classifier_model, image_to_classify):
image_to_classify = image_to_classify.permute((0, 3, 1, 2))
image_to_classify = image_to_classify.permute(0, 3, 1, 2)
transform = transforms.Compose([
transforms.Resize(size=(640, 640))