PySceneDetect/api_test.py at master · elcombato/PySceneDetect · GitHub

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

#
# PySceneDetect v0.5 API Test Script
#
# NOTE: This file can only be used with development versions of PySceneDetect,
#       and gives a high-level overview of how the new API will look and work.
#       This file is for development and testing purposes mostly, although it
#       also serves as a base for further example and test programs.
#

from __future__ import print_function

import scenedetect
import scenedetect.detectors
import scenedetect.manager


def main():

    print("Running PySceneDetect API test...")

    print("PySceneDetect version being used: %s" % str(scenedetect.__version__))

    content_detector = scenedetect.detectors.ContentDetector()
    smgr = scenedetect.manager.SceneManager(detector = content_detector)
    scenedetect.detect_scenes_file("goldeneye.mp4", smgr)
    print("Detected %d scenes in video (algorithm = content, threshold = default)." % (len(smgr.scene_list)))

    content_detector = scenedetect.detectors.ContentDetector(threshold = 27)
    smgr = scenedetect.manager.SceneManager(detector = content_detector, downscale_factor = 2)
    scenedetect.detect_scenes_file("goldeneye.mp4", smgr)
    print("Detected %d scenes in video (algorithm = content, threshold = 27)." % (len(smgr.scene_list)))

    threshold = scenedetect.detectors.ThresholdDetector(threshold = 100)
    smgr = scenedetect.manager.SceneManager(detector = threshold, perf_update_rate = 5)
    scenedetect.detect_scenes_file("goldeneye.mp4", smgr)
    print("Detected %d scenes in video (algorithm = threshold, threshold = 100)." % (len(smgr.scene_list)))


if __name__ == "__main__":
    main()