Commit 4c71282e 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


Former-commit-id: 763b0b70
parent 65bcdab0
......@@ -19,6 +19,7 @@ import os
import pandas as pd
import regex as re
import seaborn as sns
import tensorflow as tf
import warnings
# Ignore warning with no downstream effect
......@@ -1051,7 +1052,7 @@ def rule_based_video(
recog_limit,
coordinates._arena,
detection_mode=coordinates._arena_detection,
cnn_model=self._ellipse_detection_model,
cnn_model=coordinates._ellipse_detection_model,
)
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