fix:修复视频生成错误

This commit is contained in:
黄建武
2025-05-09 15:38:31 +08:00
parent 8c0f637ab1
commit 1888849270

View File

@@ -28,8 +28,8 @@ class VideoReader:
self.save_quality = save_quality
self.frame_dir = frame_dir or get_app_dir("output_frames")
self.grid_dir = grid_dir or get_app_dir("grid_output")
print(f"视频路径:{video_path}",self.frame_dir,self.grid_dir)
self.font_path = font_path
self.grid_dir = grid_dir
def format_time(self, seconds: float) -> str:
mm = int(seconds // 60)
@@ -107,18 +107,21 @@ class VideoReader:
logger.info("🚀 开始提取视频帧...")
try:
# 确保目录存在
print(self.frame_dir,self.grid_dir)
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_"):
os.remove(os.path.join(self.frame_dir, file))
print(self.frame_dir,self.grid_dir)
#清空网格文件夹
for file in os.listdir(self.grid_dir):
if file.startswith("grid_"):
os.remove(os.path.join(self.grid_dir, file))
print(self.frame_dir,self.grid_dir)
self.extract_frames()
print("2#3",self.frame_dir,self.grid_dir)
logger.info("🧩 开始拼接网格图...")
image_paths = []
groups = self.group_images()