@AlextheEngineer It is a great project! Thank you! I'm not familiar with rendering and pytorch3d, is there a way to render mesh on the original image?