Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Lucas Miranda
deepOF
Commits
41e55045
Commit
41e55045
authored
Mar 07, 2021
by
lucas_miranda
Browse files
Added nose2body to rule_based_annotation()
parent
5c7272c8
Changes
1
Hide whitespace changes
Inline
Side-by-side
deepof/pose_utils.py
View file @
41e55045
...
...
@@ -836,6 +836,13 @@ def tag_rulebased_frames(
"""Partial closure over cv2.putText to avoid code repetition"""
return
cv2
.
putText
(
frame
,
text
,
pos
,
font
,
0.75
,
col
,
2
)
def
conditional_flag
():
"""Returns a tag depending on a condition"""
if
frame_speeds
[
animal_ids
[
0
]]
>
frame_speeds
[
animal_ids
[
1
]]:
return
left_flag
else
:
return
right_flag
def
conditional_pos
():
"""Returns a position depending on a condition"""
if
frame_speeds
[
animal_ids
[
0
]]
>
frame_speeds
[
animal_ids
[
1
]]:
...
...
@@ -860,6 +867,11 @@ def tag_rulebased_frames(
)
)
# Keep track of space usage in the output video
# The flags are set to False as soon as the lower
# corners are occupied with text
left_flag
,
right_flag
=
True
,
True
if
debug
:
# Print arena for debugging
cv2
.
ellipse
(
frame
,
arena
[
0
],
arena
[
1
],
arena
[
2
],
0
,
360
,
(
0
,
255
,
0
),
3
)
...
...
@@ -880,28 +892,49 @@ def tag_rulebased_frames(
if
len
(
animal_ids
)
>
1
:
if
tag_dict
[
"nose2nose"
][
fnum
]
and
not
tag_dict
[
"sidebyside"
][
fnum
]
:
if
tag_dict
[
"nose2nose"
][
fnum
]:
write_on_frame
(
"Nose-Nose"
,
conditional_pos
())
if
(
tag_dict
[
animal_ids
[
0
]
+
"_nose2tail"
][
fnum
]
and
not
tag_dict
[
"sidereside"
][
fnum
]
):
if
frame_speeds
[
animal_ids
[
0
]]
>
frame_speeds
[
animal_ids
[
1
]]:
left_flag
=
False
else
:
right_flag
=
False
if
tag_dict
[
animal_ids
[
0
]
+
"_nose2body"
][
fnum
]
and
left_flag
:
write_on_frame
(
"nose2body"
,
corners
[
"downleft"
])
left_flag
=
False
if
tag_dict
[
animal_ids
[
1
]
+
"_nose2body"
][
fnum
]
and
right_flag
:
write_on_frame
(
"nose2body"
,
corners
[
"downright"
])
right_flag
=
False
if
tag_dict
[
animal_ids
[
0
]
+
"_nose2tail"
][
fnum
]
and
left_flag
:
write_on_frame
(
"Nose-Tail"
,
corners
[
"downleft"
])
if
(
tag_dict
[
animal_ids
[
1
]
+
"_nose2tail"
][
fnum
]
and
not
tag_dict
[
"sidereside"
][
fnum
]
):
left_flag
=
False
if
tag_dict
[
animal_ids
[
1
]
+
"_nose2tail"
][
fnum
]
and
right_flag
:
write_on_frame
(
"Nose-Tail"
,
corners
[
"downright"
])
if
tag_dict
[
"sidebyside"
][
fnum
]:
right_flag
=
False
if
tag_dict
[
"sidebyside"
][
fnum
]
and
left_flag
and
conditional_flag
():
write_on_frame
(
"Side-side"
,
conditional_pos
(),
)
if
tag_dict
[
"sidereside"
][
fnum
]:
if
frame_speeds
[
animal_ids
[
0
]]
>
frame_speeds
[
animal_ids
[
1
]]:
left_flag
=
False
else
:
right_flag
=
False
if
tag_dict
[
"sidereside"
][
fnum
]
and
left_flag
and
conditional_flag
():
write_on_frame
(
"Side-Rside"
,
conditional_pos
(),
)
if
frame_speeds
[
animal_ids
[
0
]]
>
frame_speeds
[
animal_ids
[
1
]]:
left_flag
=
False
else
:
right_flag
=
False
# for _id, down_pos, up_pos in zipped_pos:
# if (
# tag_dict[_id + "_following"][fnum]
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment