Commit be629717 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added hyperparameter logging to tests

parent a2278c5f
Pipeline #93083 passed with stage
in 26 minutes and 31 seconds
......@@ -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_COMPLEX_SMALL
font = cv2.FONT_HERSHEY_DUPLEX
# Capture speeds
try:
......
......@@ -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,
......
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