|
import gradio as gr |
|
|
|
from scenedetect import open_video, SceneManager |
|
from scenedetect.detectors import ContentDetector |
|
|
|
|
|
from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip |
|
|
|
outputs = [] |
|
|
|
def find_scenes(video_path, threshold=27.0): |
|
|
|
video = open_video(video_path) |
|
scene_manager = SceneManager() |
|
scene_manager.add_detector( |
|
ContentDetector(threshold=threshold)) |
|
scene_manager.detect_scenes(video, show_progress=True) |
|
scene_list = scene_manager.get_scene_list() |
|
|
|
outputs.append("json") |
|
|
|
shots = [] |
|
|
|
for i, scene in enumerate(scene_list): |
|
shot_in = scene[0].get_frames() / scene[0].get_framerate() |
|
shot_out = (scene[1].get_frames() - 1) / scene[0].get_framerate() |
|
target_name = str(i)+"_cut.mp4" |
|
ffmpeg_extract_subclip(video_path, shot_in, shot_out, targetname=target_name) |
|
shots.append(target_name) |
|
outputs.append("video") |
|
|
|
|
|
|
|
|
|
print(scene_list, *tuple(shots), sep=', ') |
|
print(outputs) |
|
|
|
|
|
|
|
return scene_list, tuple(shots) |
|
|
|
video_input=gr.Video(source="upload", format="mp4"); |
|
|
|
gr.Interface(fn=find_scenes, inputs=video_input, outputs=outputs).launch() |