Visualize results

We should write a method that is able to take an output from the training and compare it to the ground truth image. Something that works like [MATLAB's imshowpair](https://www.mathworks.com/help/images/ref/imshowpair.html).

