nodes/nomadoor-ComfyUI-Temporal-Mask-Tools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: replace view with reshape for tensor manipulation in _prune_duration function

Browse Source
This commit is contained in:
nomadoor
2025-10-17 19:50:00 +09:00
parent 01d633e021
commit bbbda337a9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/temporal_mask_remove_short_objects.py
+2 -2
View File
@@ -73,7 +73,7 @@ def _prune_duration(mask_bool: Tensor, min_duration: int) -> Tensor:
if min_duration <= 1:
return mask_bool
batch, frames, height, width = mask_bool.shape
flat = mask_bool.view(batch, frames, -1)
flat = mask_bool.reshape(batch, frames, -1)
lengths = torch.zeros_like(flat, dtype=torch.int32)
running = torch.zeros((batch, flat.shape[2]), dtype=torch.int32, device=mask_bool.device)
for t in range(frames):
@@ -86,7 +86,7 @@ def _prune_duration(mask_bool: Tensor, min_duration: int) -> Tensor:
running = torch.where(active, running + 1, torch.zeros_like(running))
lengths[:, t, :] += running - 1
keep = lengths >= min_duration
return (flat & keep).view(batch, frames, height, width)
return (flat & keep).reshape(batch, frames, height, width)
class TemporalMaskRemoveShortObjects(io.ComfyNode):