bowong
/
ComfyUI-CustomNode
6
0
Fork 0
Code Pull Requests Releases Activity

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

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
__init__.py
View File

@ -54,7 +54,7 @@ class FaceDetect:
config = {"start": start, "end": end}
else:
raise RuntimeError("未找到符合要求的视频片段")
return (image, image_selected, cls, prob, nums, str(period), json.dumps(config), start, end - start)
return (image, image_selected, cls, prob, nums, str(period), json.dumps(config), start, end - start+1)
class FaceExtract:

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)