Skip to content

Commit b66d897

Browse files
committed
[detectors] Implement Koala-36M
Work in progress. #441
1 parent 2c55a55 commit b66d897

File tree

3 files changed

+17
-0
lines changed

3 files changed

+17
-0
lines changed

scenedetect/_cli/__init__.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@
4242
ContentDetector,
4343
HashDetector,
4444
HistogramDetector,
45+
KoalaDetector,
4546
ThresholdDetector,
4647
)
4748
from scenedetect.platform import get_cv2_imwrite_params, get_system_version_info
@@ -1577,3 +1578,16 @@ def save_qp_command(
15771578
scenedetect.add_command(list_scenes_command)
15781579
scenedetect.add_command(save_images_command)
15791580
scenedetect.add_command(split_video_command)
1581+
1582+
1583+
@click.command("detect-koala", cls=Command, help="""WIP""")
1584+
@click.pass_context
1585+
def detect_koala_command(
1586+
ctx: click.Context,
1587+
):
1588+
ctx = ctx.obj
1589+
assert isinstance(ctx, CliContext)
1590+
ctx.add_detector(KoalaDetector, {"min_scene_len": None})
1591+
1592+
1593+
scenedetect.add_command(detect_koala_command)

scenedetect/detectors/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040
from scenedetect.detectors.adaptive_detector import AdaptiveDetector
4141
from scenedetect.detectors.hash_detector import HashDetector
4242
from scenedetect.detectors.histogram_detector import HistogramDetector
43+
from scenedetect.detectors.koala_detector import KoalaDetector
4344

4445
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
4546
# #

tests/test_detectors.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
ContentDetector,
3030
HashDetector,
3131
HistogramDetector,
32+
KoalaDetector,
3233
ThresholdDetector,
3334
)
3435

@@ -37,6 +38,7 @@
3738
ContentDetector,
3839
HashDetector,
3940
HistogramDetector,
41+
KoalaDetector,
4042
)
4143

4244
ALL_DETECTORS: ty.Tuple[ty.Type[SceneDetector]] = (*FAST_CUT_DETECTORS, ThresholdDetector)

0 commit comments

Comments
 (0)