Commit 763b0b70 authored by lucas_miranda's avatar lucas_miranda
Browse files

Expanded circular_arena_recognition to detect the median arena across n frames...

Expanded circular_arena_recognition to detect the median arena across n frames (the minimum between 100 and the length of the video, by default) using a cnn
parent 110ce850
Pipeline #97204 canceled with stages
in 1 minute and 35 seconds
...@@ -19,6 +19,7 @@ import os ...@@ -19,6 +19,7 @@ import os
import pandas as pd import pandas as pd
import regex as re import regex as re
import seaborn as sns import seaborn as sns
import tensorflow as tf
import warnings import warnings
# Ignore warning with no downstream effect # Ignore warning with no downstream effect
...@@ -1051,7 +1052,7 @@ def rule_based_video( ...@@ -1051,7 +1052,7 @@ def rule_based_video(
recog_limit, recog_limit,
coordinates._arena, coordinates._arena,
detection_mode=coordinates._arena_detection, detection_mode=coordinates._arena_detection,
cnn_model=self._ellipse_detection_model, cnn_model=coordinates._ellipse_detection_model,
) )
corners = frame_corners(h, w) corners = frame_corners(h, w)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment