Update detect.py

shadowcz007 · Mar 29, 2024 · c4be286 · c4be286
1 parent b4c304c
commit c4be286
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nodes/detect.py b/nodes/detect.py
@@ -193,8 +193,8 @@ def run(self,image,confidence,model,type="YOLO-World",target_label="",debug="on"
             # 创建一个黑色图
             mask = Image.new("L", image.size)
             mask=pil2tensor(mask)
-            masks_total.append(mask)
-            grids_total.append((0,0,image.size[0],image.size[1]))
+            masks_total.append([mask])
+            grids_total.append([(0,0,image.size[0],image.size[1])])
             names_total.append(['-'])
             # masks = result.masks  # Masks object for segmentation masks outputs
             # keypoints = result.keypoints  # Keypoints object for pose outputs