comi-logic/ComfyUI-CustomNode
8
0
Fork 0
Code Pull Requests Releases Activity

PERF 修改视频切片长度为输入长度

Browse Source
This commit is contained in:
康宇佳
2025-02-19 16:23:43 +08:00
parent 6f83efdd94
commit d4978b8a4a
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test_single_image.py
View File

@@ -96,7 +96,13 @@ def test_node(image:torch.Tensor,length=10,thres=95,model_name="convnext_tiny"):
end = nums[idx]
if end - start + 1 >= length:
period.append([start, end])
return (image.permute(0,2,3,1), image.permute(0,2,3,1)[nums,:,:,:], str(preds), str(probs), str(nums), period)
temp_period = []
for i in period:
a = i[0]
while a+length-1 <= i[1]:
temp_period.append([a,a+length-1])
a = a+length
return (image.permute(0,2,3,1), image.permute(0,2,3,1)[nums,:,:,:], str(preds), str(probs), str(nums), temp_period)