diff --git a/Dockerfile b/Dockerfile index 8430d80..6031e6e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -52,6 +52,7 @@ WORKDIR /app # 复制二进制和运行时必需的文件 COPY --from=builder /build/main . COPY --from=builder /build/config.yml . +COPY --from=builder /build/scripts ./scripts EXPOSE 3010 CMD ["./main"] diff --git a/scripts/scene_detect.py b/scripts/scene_detect.py new file mode 100644 index 0000000..2e2c256 --- /dev/null +++ b/scripts/scene_detect.py @@ -0,0 +1,45 @@ +#!/usr/bin/env python3 +"""PySceneDetect 场景检测脚本,由 Go 服务调用。""" + +import argparse +import json +import sys + +from scenedetect import open_video, SceneManager +from scenedetect.detectors import ContentDetector + + +def detect_scenes(video_path: str, threshold: float) -> list[dict]: + """检测视频场景切分点,返回场景时间边界列表。""" + video = open_video(video_path) + scene_manager = SceneManager() + scene_manager.add_detector(ContentDetector(threshold=threshold)) + scene_manager.detect_scenes(video) + scene_list = scene_manager.get_scene_list() + + scenes = [] + for start, end in scene_list: + scenes.append({ + "start_time": start.seconds, + "end_time": end.seconds, + }) + return scenes + + +def main(): + parser = argparse.ArgumentParser(description="视频场景检测") + parser.add_argument("--input", required=True, help="输入视频路径") + parser.add_argument("--output", required=True, help="输出 JSON 文件路径") + parser.add_argument("--threshold", type=float, default=27.0, help="内容检测阈值") + args = parser.parse_args() + + scenes = detect_scenes(args.input, args.threshold) + + with open(args.output, "w") as f: + json.dump(scenes, f, indent=2) + + print(f"检测完成: {len(scenes)} 个场景", file=sys.stderr) + + +if __name__ == "__main__": + main()