Improving ML performance with image data