ML-assistant Imaging