Spaces:
Sleeping
Sleeping
Update video_processing.py
Browse files- video_processing.py +7 -3
video_processing.py
CHANGED
@@ -66,15 +66,19 @@ def download_video(url):
|
|
66 |
def sanitize_filename(filename):
|
67 |
return "".join([c if c.isalnum() or c in " .-_()" else "_" for c in filename])
|
68 |
|
69 |
-
def find_scenes(video_path):
|
70 |
video_manager = VideoManager([video_path])
|
71 |
scene_manager = SceneManager()
|
72 |
-
scene_manager.add_detector(ContentDetector(threshold=33))
|
73 |
-
|
|
|
|
|
74 |
video_manager.start()
|
|
|
75 |
scene_manager.detect_scenes(frame_source=video_manager)
|
76 |
scene_list = scene_manager.get_scene_list()
|
77 |
video_manager.release()
|
|
|
78 |
scenes = [(start.get_timecode(), end.get_timecode()) for start, end in scene_list]
|
79 |
return scenes
|
80 |
|
|
|
66 |
def sanitize_filename(filename):
|
67 |
return "".join([c if c.isalnum() or c in " .-_()" else "_" for c in filename])
|
68 |
|
69 |
+
def find_scenes(video_path, sample_rate=3, downscale_factor=2):
|
70 |
video_manager = VideoManager([video_path])
|
71 |
scene_manager = SceneManager()
|
72 |
+
scene_manager.add_detector(ContentDetector(threshold=33))
|
73 |
+
|
74 |
+
video_manager.set_downscale_factor(downscale_factor)
|
75 |
+
video_manager.set_duration(start_time=None, end_time=None, frame_skip=sample_rate) # Skip frames
|
76 |
video_manager.start()
|
77 |
+
|
78 |
scene_manager.detect_scenes(frame_source=video_manager)
|
79 |
scene_list = scene_manager.get_scene_list()
|
80 |
video_manager.release()
|
81 |
+
|
82 |
scenes = [(start.get_timecode(), end.get_timecode()) for start, end in scene_list]
|
83 |
return scenes
|
84 |
|