From 3100f0cea003f6b6eea397fda7f0f39f40fc6ef3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=96=8C?= <259278618@qq.com>
Date: Mon, 22 Jun 2026 16:22:17 +0800
Subject: [PATCH] =?UTF-8?q?docker=E5=AE=89=E8=A3=85FFmpeg=20+=20Python3=20?=
 =?UTF-8?q?+=20PySceneDetect=20+=20Node.js=20+=20HyperFrames?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile              |  1 +
 scripts/scene_detect.py | 45 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+)
 create mode 100644 scripts/scene_detect.py

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()