diff --git a/sciencebeam_gym/trainer/evaluator.py b/sciencebeam_gym/trainer/evaluator.py index 28edd0b0c44bcff49cfbae49031c313d1dd1e130..ec7763e92fc627a78c2f5b6ac67cf222a7d8ac6b 100644 --- a/sciencebeam_gym/trainer/evaluator.py +++ b/sciencebeam_gym/trainer/evaluator.py @@ -33,7 +33,7 @@ def plot_image(ax, image, label): ax.imshow(image, aspect='auto') else: ax.imshow(np.dstack((image.astype(np.uint8),)*3)*100, aspect='auto') - ax.set_title(label, color=(0.5, 0.5, 0.5), y=0.99) + ax.set_title(label, color=(0.5, 0.5, 0.5), y=0.995) ax.set_axis_off() ax.set(xlim=[0, 255], ylim=[255, 0], aspect=1) ax.axes.get_xaxis().set_visible(False)