Imported from https://github.com/natowi/3D-Reconstruction-with-Neural-Network