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
be629717
Commit
be629717
authored
Feb 09, 2021
by
lucas_miranda
Browse files
Added hyperparameter logging to tests
parent
a2278c5f
Pipeline
#93083
passed with stage
in 26 minutes and 31 seconds
Changes
2
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
deepof/pose_utils.py
View file @
be629717
...
...
@@ -626,7 +626,7 @@ def tag_rulebased_frames(
def
write_on_frame
(
text
,
pos
,
col
=
(
255
,
255
,
255
)):
"""Partial closure over cv2.putText to avoid code repetition"""
return
cv2
.
putText
(
frame
,
text
,
pos
,
font
,
1
,
col
,
2
)
return
cv2
.
putText
(
frame
,
text
,
pos
,
font
,
0.75
,
col
,
2
)
def
conditional_pos
():
"""Returns a position depending on a condition"""
...
...
@@ -672,7 +672,7 @@ def tag_rulebased_frames(
thickness
=-
1
,
)
# Print frame number
write_on_frame
(
"Frame "
+
str
(
fnum
),
corners
[
"downleft"
]
)
write_on_frame
(
"Frame "
+
str
(
fnum
),
(
int
(
w
*
0.3
/
10
),
int
(
h
/
1.15
))
)
if
tag_dict
[
"nose2nose"
][
fnum
]
and
not
tag_dict
[
"sidebyside"
][
fnum
]:
write_on_frame
(
"Nose-Nose"
,
conditional_pos
())
...
...
@@ -809,7 +809,7 @@ def rule_based_video(
print
(
"Can't receive frame (stream end?). Exiting ..."
)
break
font
=
cv2
.
FONT_HERSHEY_
COM
PLEX
_SMALL
font
=
cv2
.
FONT_HERSHEY_
DU
PLEX
# Capture speeds
try
:
...
...
tests/test_train_utils.py
View file @
be629717
...
...
@@ -127,6 +127,7 @@ def test_autoencoder_fitting(
batch_size
=
batch_size
,
encoding_size
=
encoding_size
,
epochs
=
1
,
log_hparams
=
True
,
n_components
=
k
,
loss
=
loss
,
phenotype_class
=
pheno_class
,
...
...
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