Patches CNN generator and updates README

The CNN generator had hard coded input dimensionality based on the example. This is patched by adding function arguments image_height and image_width.