feat(#6): add example

williamfzc · Jul 23, 2019 · 97ceeb9 · 97ceeb9
1 parent 7ff66f5
commit 97ceeb9
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -16,3 +16,4 @@ script:
   - python classify_with_svm.py
   - python cut_and_classify.py
   - python multi_video.py
+  - python mini.py
diff --git a/example/mini.py b/example/mini.py
@@ -0,0 +1,21 @@
+from stagesepx.cutter import VideoCutter
+from stagesepx.classifier import SSIMClassifier
+from stagesepx.reporter import Reporter
+
+# cut
+video_path = '../test.mp4'
+cutter = VideoCutter()
+res = cutter.cut(video_path)
+stable = res.get_stable_range()
+
+# classify
+cl = SSIMClassifier()
+cl.load(stable)
+
+res = cl.classify(
+    video_path,
+    stable,
+)
+
+# draw
+Reporter.draw(res)