nodes/ltdrdata-ComfyUI-Impact-Pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed: SegmDetectorCombined - incorrect empty mask dimension (#1021)

Browse Source
This commit is contained in:
NicolasKlenert
2025-06-16 12:34:56 +09:00
committed by GitHub
parent ac3668d946
commit 6ccf9bab68
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/impact/detectors.py
+2 -2
View File
@@ -163,7 +163,7 @@ class SegmDetectorCombined:
mask = segm_detector.detect_combined(image, threshold, dilation)
if mask is None:
mask = torch.zeros((image.shape[2], image.shape[1]), dtype=torch.float32, device="cpu")
mask = torch.zeros((image.shape[1], image.shape[2]), dtype=torch.float32, device="cpu")
return (mask.unsqueeze(0),)
@@ -183,7 +183,7 @@ class BboxDetectorCombined(SegmDetectorCombined):
mask = bbox_detector.detect_combined(image, threshold, dilation)
if mask is None:
mask = torch.zeros((image.shape[2], image.shape[1]), dtype=torch.float32, device="cpu")
mask = torch.zeros((image.shape[1], image.shape[2]), dtype=torch.float32, device="cpu")
return (mask.unsqueeze(0),)