Merge pull request #340 from gnes-ai/set_max_shots_num

fix(shot-detector): rename shot detector

Author: mergify[bot]

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',

gnes/preprocessor/video/shotdetect.py renamed to 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,

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

tests/yaml/preprocessor-shotdetect_histogram.yml

@@ -1,4 +1,4 @@
-!ShotDetectPreprocessor
+!ShotDetectorPreprocessor
 parameters:
   descriptor: "block_hsv_histogram"
   distance_metric: "bhattacharya"