Commit 8d6751f8 authored by lucas_miranda's avatar lucas_miranda
Browse files

Refactored pose_utils.py and pose_utils.py

parent b91bf8c1
Pipeline #83094 canceled with stage
in 13 minutes and 2 seconds
......@@ -376,6 +376,7 @@ def get_hparameters(hparams: dict = {}) -> dict:
"huddle_forward": 15,
"huddle_spine": 10,
"huddle_speed": 1,
"fps": 0,
}
for k, v in hparams.items():
......@@ -555,7 +556,6 @@ def rule_based_video(
tag_dict,
mode,
path,
fps,
frame_limit,
recog_limit,
hparams,
......@@ -802,7 +802,7 @@ def rule_based_video(
writer.open(
re.findall("(.*?)_", tracks[vid_index])[0] + "_tagged.avi",
cv2.VideoWriter_fourcc(*"MJPG"),
(fps if fps != 0 else cv2.CAP_PROP_FPS),
(hparams["fps"] if hparams["fps"] != 0 else cv2.CAP_PROP_FPS),
(frame.shape[1], frame.shape[0]),
True,
)
......
......@@ -51,7 +51,7 @@ class project:
smooth_alpha: float = 0.1,
arena_dims: tuple = (1,),
model: str = "mouse_topview",
animal_ids: List = (""),
animal_ids: List = tuple([""]),
):
self.path = path
......@@ -368,7 +368,7 @@ class coordinates:
exp_conditions: dict = None,
distances: dict = None,
angles: dict = None,
animal_ids: List = (""),
animal_ids: List = tuple([""]),
):
self._tables = tables
self.distances = distances
......
Supports Markdown
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