Return prediction as image vector

Worked with vector image, but not flexible enough. Need to make image data persistent!