Rough Surface Reconstruction with Machine Learning Methods