There are lots of ways to go about this. Here's one.
In this exercise, we start to define and place the images in the figure. To fully define the position and size of the input image, we need to provide the x- and y-position of the lower left corner (a.k.a. "left" and "bottom" of the image) and its height and width. If we assume that it's left position is determined by the left border, this exercise fills in the rest of what we need.
Feel free to use my solution to the last exercise as a starting point: nn_viz_02.py
Ready? Go for it.
2 comments