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

FIX 修复精确裁剪视频节点

This commit is contained in:
kyj@bowong.ai 2025-03-26 15:48:59 +08:00
parent 0c2d564b55
commit bb89d4bc41
1 changed files with 164 additions and 157 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes/videocut.py
View File

@ -6,7 +6,6 @@ import uuid
from datetime import datetime from datetime import datetime
import ffmpy import ffmpy
import torchaudio
import torchvision.io import torchvision.io
video_extensions = ['webm', 'mp4', 'mkv', 'gif', 'mov'] video_extensions = ['webm', 'mp4', 'mkv', 'gif', 'mov']
@ -36,6 +35,7 @@ class VideoCut:
CATEGORY = "不忘科技-自定义节点🚩" CATEGORY = "不忘科技-自定义节点🚩"
def cut(self, video_path, start, end): def cut(self, video_path, start, end):
try:
# 原文件名 # 原文件名
origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1]) origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1])
# 新文件名 复制改名适配ffmpeg # 新文件名 复制改名适配ffmpeg
@ -118,6 +118,9 @@ class VideoCut:
except: except:
pass pass
return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},) return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},)
except:
traceback.print_exc()
raise Exception("Cut Failed")
class VideoCutByFramePoint: class VideoCutByFramePoint:
"""FFMPEG视频剪辑-帧位""" """FFMPEG视频剪辑-帧位"""
@ -144,6 +147,7 @@ class VideoCutByFramePoint:
CATEGORY = "不忘科技-自定义节点🚩" CATEGORY = "不忘科技-自定义节点🚩"
def cut(self, video_path, start_point, duration, fps, force_match_fps): def cut(self, video_path, start_point, duration, fps, force_match_fps):
try:
# 原文件名 # 原文件名
origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1]) origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1])
# 新文件名 复制改名适配ffmpeg # 新文件名 复制改名适配ffmpeg
@ -228,3 +232,6 @@ class VideoCutByFramePoint:
except: except:
pass pass
return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},) return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},)
except:
traceback.print_exc()
raise Exception("Cut Failed")