2020_CaricatureFace_code

 

网格处理的库，对ply，obj，off文件处理为np文件，再把np文件转换为mesh

https://pypi.org/project/openmesh/，这个很好安装，pip install openmesh就可以

https://github.com/MPI-IS/mesh，马普所的，需要编译

我是在win下测试的，需要的环境pytorch，python3.7，openmesh，opencv

 

训练时需要的数据(1)image.jpg (2)landmark.npy (3)mesh_gt.npy ，这部分数据没有开源

测试：

• 提供了model,data,exp文件夹，可直接运行，输入imge和landmark(作为gt用来计算error)，输出mesh_npy文件和对应的landmark。
• 测试的输入数据再exp文件夹下，txt是路径；输出到record文件夹(自己创建的)，lrecord是计算的landmark_npy，vrecord是生成的mesh顶点npy文件。
• loadmodel时有一个小报错，在cariface.py下，145行ck1 = torch.load(model1_path)，默认使用的cuda1，改为                      ck1 = torch.load(model1_path, map_location