3D人脸重建

Halcom

【1】阿里的HRN：https://github.com/youngLBW/HRN
【2】微软的：https://github.com/sicxu/Deep3DFaceRecon_pytorch
环境构建：HRN人脸重建的依赖包版本号 - 哔哩哔哩
报错No module named 'nvdiffrast'：HRN环境构建说明 - 哔哩哔哩

1. # from modelscope import snapshot_download
   
2. # model_dir = snapshot_download('iic/cv_HRN_head-reconstruction')
   
3. 
   
4. import os
   
5. import cv2
   
6. from modelscope.models.cv.face_reconstruction.utils import write_obj
   
7. from modelscope.outputs import OutputKeys
   
8. from modelscope.pipelines import pipeline
   
9. from modelscope.utils.constant import Tasks
   
10. 
    
11. def save_results(result, save_root):
    
12.     os.makedirs(save_root, exist_ok=True)
    
13. 
    
14.     # export obj and texture
    
15.     mesh = result[OutputKeys.OUTPUT]['mesh']
    
16.     texture_map = result[OutputKeys.OUTPUT_IMG]
    
17.     mesh['texture_map'] = texture_map
    
18.     write_obj(os.path.join(save_root, 'head_recon_result.obj'), mesh)
    
19. 
    
20.     print(f'Output written to {os.path.abspath(save_root)}')
    
21. 
    
22. head_reconstruction = pipeline(Tasks.head_reconstruction, model='./HRN/master', hair_tex=False)
    
23. result = head_reconstruction('D:/2-LearningCode/19_3Dface/guma.JPG')
    
24. 
    
25. save_results(result, './HRN')
    

复制代码