From 763b0b7035ef257c353e87e66a4a7b03bb9de422 Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Tue, 30 Mar 2021 15:48:24 +0200
Subject: [PATCH] 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
---
deepof/pose_utils.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/deepof/pose_utils.py b/deepof/pose_utils.py
index 08858ff8..5deb463b 100644
--- a/deepof/pose_utils.py
+++ b/deepof/pose_utils.py
@@ -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)
--
GitLab