hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix the bug where the video is incorrectly scaled

Browse Source
This commit is contained in:
harry
2024-04-03 19:01:40 +08:00
parent b94cd353ba
commit fd81698f66
1 changed files with 41 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
app/services/video.py
View File

@@ -64,24 +64,34 @@ def combine_videos(combined_video_path: str,
clip = clip.set_fps(30) clip = clip.set_fps(30)
# Not all videos are same size, so we need to resize them # Not all videos are same size, so we need to resize them
# logger.info(f"{video_path}: size is {clip.w} x {clip.h}, expected {video_width} x {video_height}") clip_w, clip_h = clip.size
if clip.w != video_width or clip.h != video_height: if clip_w != video_width or clip_h != video_height:
if round((clip.w / clip.h), 4) < 0.5625: clip_ratio = clip.w / clip.h
clip = crop(clip, video_ratio = video_width / video_height
width=clip.w,
height=round(clip.w / 0.5625), if clip_ratio == video_ratio:
x_center=clip.w / 2, # 等比例缩放
y_center=clip.h / 2 clip = clip.resize((video_width, video_height))
)
else: else:
clip = crop(clip, # 等比缩放视频
width=round(0.5625 * clip.h), if clip_ratio > video_ratio:
height=clip.h, # 按照目标宽度等比缩放
x_center=clip.w / 2, scale_factor = video_width / clip_w
y_center=clip.h / 2 else:
) # 按照目标高度等比缩放
logger.info(f"resizing video to {video_width} x {video_height}, clip size: {clip.w} x {clip.h}") scale_factor = video_height / clip_h
clip = clip.resize((video_width, video_height))
new_width = int(clip_w * scale_factor)
new_height = int(clip_h * scale_factor)
clip_resized = clip.resize(newsize=(new_width, new_height))
background = ColorClip(size=(video_width, video_height), color=(0, 0, 0))
clip = CompositeVideoClip([
background.set_duration(clip.duration),
clip_resized.set_position("center")
])
logger.info(f"resizing video to {video_width} x {video_height}, clip size: {clip_w} x {clip_h}")
if clip.duration > max_clip_duration: if clip.duration > max_clip_duration:
clip = clip.subclip(0, max_clip_duration) clip = clip.subclip(0, max_clip_duration)
@@ -263,6 +273,20 @@ if __name__ == "__main__":
audio_file = f"{task_dir}/audio.mp3" audio_file = f"{task_dir}/audio.mp3"
subtitle_file = f"{task_dir}/subtitle.srt" subtitle_file = f"{task_dir}/subtitle.srt"
output_file = f"{task_dir}/final.mp4" output_file = f"{task_dir}/final.mp4"
video_paths = []
for file in os.listdir(utils.storage_dir("test")):
if file.endswith(".mp4"):
video_paths.append(os.path.join(task_dir, file))
combine_videos(combined_video_path=video_file,
audio_file=audio_file,
video_paths=video_paths,
video_aspect=VideoAspect.portrait,
video_concat_mode=VideoConcatMode.random,
max_clip_duration=5,
threads=2)
cfg = VideoParams() cfg = VideoParams()
cfg.video_aspect = VideoAspect.portrait cfg.video_aspect = VideoAspect.portrait
cfg.font_name = "STHeitiMedium.ttc" cfg.font_name = "STHeitiMedium.ttc"