Spaces:
Runtime error
Runtime error
File size: 2,466 Bytes
6ed89af |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from PIL import Image, ImageFilter
import random
import os
from pathlib import Path
import cv2
import numpy as np
class VideoMergerWithSmoothTransition:
def __init__(self, fps: int = 30, transition_duration: float = 0.5):
self.fps = fps
self.transition_duration = transition_duration
def merge_videos(self, input_folder: str, output_filename: str):
video_files = [f for f in Path(input_folder).glob("*.mp4")]
if not video_files:
print("No video files found in the specified directory.")
return
videos = []
for video_file in video_files:
video = cv2.VideoCapture(str(video_file))
videos.append(video)
width = int(videos[0].get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(videos[0].get(cv2.CAP_PROP_FRAME_HEIGHT))
fourcc = cv2.VideoWriter_fourcc(*'MP4V')
out = cv2.VideoWriter(output_filename, fourcc, self.fps, (width, height))
transition_frames_count = int(self.fps * self.transition_duration)
for i, video in enumerate(videos):
ret, prev_frame = video.read()
while ret:
if i < len(videos) - 1 and not video.get(cv2.CAP_PROP_POS_FRAMES) < video.get(cv2.CAP_PROP_FRAME_COUNT) - transition_frames_count:
alpha = (video.get(cv2.CAP_PROP_POS_FRAMES) - (video.get(cv2.CAP_PROP_FRAME_COUNT) - transition_frames_count)) / transition_frames_count
ret_next, next_frame = videos[i + 1].read()
if ret_next:
blended_frame = cv2.addWeighted(prev_frame, 1 - alpha, next_frame, alpha, 0)
out.write(blended_frame)
ret, prev_frame = video.read()
continue
out.write(prev_frame)
ret, prev_frame = video.read()
video.release()
out.release()
print(f"Concatenated video saved to {output_filename}.")
if __name__ == '__main__':
# 使用例 (コメントアウトされています)
merger = VideoMergerWithSmoothTransition()
input_folder_path = r"image\Echoes-of-Creation_Blurred_mov"
output_folder_path = f"{input_folder_path}_Final"
os.makedirs(output_folder_path, exist_ok=True)
output_video_path = os.path.join(output_folder_path, "concatenated_video.mp4")
merger.merge_videos(input_folder_path, output_video_path) |