Skip to content
Snippets Groups Projects
Commit 5df78695 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[script] added epoch and iteration number on the figure title

parent e1cd9520
No related branches found
No related tags found
No related merge requests found
......@@ -66,6 +66,7 @@ def main(user_input=None):
filename = os.path.join(data_dir, f)
temp = filename.split('_')
epoch = int(temp[-2])
iteration = int(temp[-1].split('.')[-2])
if epoch < start_epoch:
continue
......@@ -79,7 +80,7 @@ def main(user_input=None):
from matplotlib import pyplot
fig, axarr = pyplot.subplots(1,2)
fig.suptitle("ID = {}".format(id_example))
fig.suptitle("Epoch {}, iteration {} => ID {}".format(epoch, iteration, id_example))
axarr[0].set_title("Real pose = {}".format(real_pose))
axarr[0].imshow(numpy.rollaxis(numpy.rollaxis(real_example, 2),2))
axarr[1].set_title("Target pose = {}".format(target_pose))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment