File size: 1,714 Bytes
2252f3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from moviepy.editor import VideoFileClip, VideoClip, ImageClip, CompositeVideoClip, ImageSequenceClip
from moviepy.editor import VideoFileClip, clips_array
import os
import argparse
def write_video(vclip, fps=25, save_path='res.mp4'):
if isinstance(vclip, list):
vclip = ImageSequenceClip(vclip, fps=fps)
vclip.write_videofile(save_path, codec="libx264")
def load_video(vpath):
return VideoFileClip(vpath)
def concat_video_clips(clips, videos_per_row=3):
if isinstance(clips[0], str):
clips = [VideoFileClip(v) for v in clips]
elif not isinstance(clips[0], VideoFileClip):
print(f'Find {len(clips)} clips')
min_duration = min(clip.duration for clip in clips)
clips = [clip.subclip(0, min_duration) for clip in clips]
rows = [clips[i:i + videos_per_row] for i in range(0, len(clips), videos_per_row)]
final_clip = clips_array(rows)
return final_clip
def concat_video_and_frames(vpath, frames):
vclip1 = VideoClip(vpath)
vclip2 = ImageSequenceClip(frames, fps=25)
clips = concat_video_clips([vclip1, vclip2])
return clips
def concat_img_video(vclip, img):
if isinstance(vclip, str):
vclip = VideoFileClip(vclip)
image_clip = ImageClip(img).set_duration(vclip.duration)
image_clip = image_clip.resize(height=vclip.size[1])
total_width = vclip.size[0] + image_clip.size[0]
total_height = max(vclip.size[1], image_clip.size[1])
composite_clip = CompositeVideoClip([
image_clip.set_position('left'), vclip.set_position((image_clip.size[0], 0)), ],
size=(total_width, total_height))
return composite_clip
|