Add feature map sizes

[ogvalt]

It would be great if you'll add input and output feature map size

[lutzroeder]

Can you provide a sample file? Which file format and what do you expect to see?

[ogvalt]

For example, there is a function in keras that builds graph representation of model and saves it as image. Here an example of such image:

And I think it would be great to add such explicit size representation into your tool.
i.e.

[lutzroeder]

Got it. This will require #71.