fix(shot-detector): rename shot detector

gnes-ai · Oct 17, 2019 · a641d5c · a641d5c
1 parent d48eb53
commit a641d5c
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/gnes/preprocessor/__init__.py b/gnes/preprocessor/__init__.py
@@ -31,7 +31,7 @@
     'BaseVideoPreprocessor': 'base',
     'FFmpegPreprocessor': 'video.ffmpeg',
     'FFmpegVideoSegmentor': 'video.ffmpeg',
-    'ShotDetectPreprocessor': 'video.shotdetect',
+    'ShotDetectorPreprocessor': 'video.shot_detector',
     'VideoEncoderPreprocessor': 'video.video_encoder',
     'VideoDecoderPreprocessor': 'video.video_decoder',
     'AudioVanilla': 'audio.audio_vanilla',

diff --git a/gnes/preprocessor/video/shotdetect.py → gnes/preprocessor/video/shot_detector.py b/gnes/preprocessor/video/shotdetect.py → gnes/preprocessor/video/shot_detector.py
@@ -23,7 +23,7 @@
 from gnes.preprocessor.helper import compute_descriptor, compare_descriptor, detect_peak_boundary, compare_ecr
 
 
-class ShotDetectPreprocessor(BaseVideoPreprocessor):
+class ShotDetectorPreprocessor(BaseVideoPreprocessor):
     store_args_kwargs = True
 
     def __init__(self,

diff --git a/tests/yaml/preprocessor-shotdetect_edge.yml b/tests/yaml/preprocessor-shotdetect_edge.yml
@@ -1,8 +1,10 @@
-!ShotDetectPreprocessor
+!ShotDetectorPreprocessor
 parameters:
   descriptor: "canny_edge"
   distance_metric: "edge_change_ration"
   frame_size: "192:168"
   frame_rate: 10
+  kwargs:
+    max_shot_num: 5
 gnes_config:
   is_trained: true
diff --git a/tests/yaml/preprocessor-shotdetect_histogram.yml b/tests/yaml/preprocessor-shotdetect_histogram.yml
@@ -1,4 +1,4 @@
-!ShotDetectPreprocessor
+!ShotDetectorPreprocessor
 parameters:
   descriptor: "block_hsv_histogram"
   distance_metric: "bhattacharya"