Commit 92aeaafd authored by lucas_miranda's avatar lucas_miranda
Browse files

Added looking around as a tag for single mice

parent 99f493cc
Pipeline #93287 canceled with stage
in 17 minutes and 55 seconds
......@@ -212,6 +212,9 @@ def sniff_object(
nose, nosing = pos_dict[nose], True
if animal_id != "":
animal_id += "_"
if object == "arena":
if arena_type == "circular":
center = np.zeros(2) if centered_data else np.array(arena[0])
......
......@@ -377,8 +377,4 @@ def test_rule_based_tagging(multi_animal, video_output):
)
assert type(hardcoded_tags) == deepof.data.table_dict
assert list(hardcoded_tags.values())[0].shape[1] == (15 if multi_animal else 4)
# TODO:
# - Test if tagging is working properly!
assert list(hardcoded_tags.values())[0].shape[1] == (19 if multi_animal else 6)
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