Update video_reader.py

增加本地视频下载文件的检查
This commit is contained in:
scdotbox
2025-05-03 15:20:04 +08:00
committed by GitHub
parent c492f0780b
commit c0746aab57

View File

@@ -104,6 +104,9 @@ class VideoReader:
def run(self)->list[str]:
logger.info("🚀 开始提取视频帧...")
try:
# 确保目录存在
os.makedirs(self.frame_dir, exist_ok=True)
os.makedirs(self.grid_dir, exist_ok=True)
#清空帧文件夹
for file in os.listdir(self.frame_dir):
if file.startswith("frame_"):