Realistic mask generation for matter-wave lithography via machine learning