implement the visualization

[c-schicho]

Todos:

• [ ] visualise image patches (according to paper)
• [ ] move visualisation into separate function.
• [ ] maybe visualise each layer separately for better interpretations
• [ ] clean up the repo

@SeskoDk would that be ok for you? Otherwise just adapt the comment.