进行推理时,需要编辑run_inference.py中的代码。为了避免人物肖像的版权问题,文章中使用的图片不是由FaceChain实际生成的图片,特此说明。
1. 以下代码适用于Linux系统,如果希望在Windows系统中运行,
folder_path = f"/mnt/workspace/new_facechain/facechain/styles/{base_model['name']}"
需要改为
folder_path = f"styles/{base_model['name']}"
2. 如果是用女性图片进行训练,做推理的时候需要将以下代码
pose_image = 'poses/man/pose1.png'
修改为
pose_image = 'poses/woman/pose1.png'
3.缺省是使用第一个人物风格,
如果希望选择其他人物风格,例如
则需要找到对应的人物风格索引(以上人物风格索引为1),并修改代码
style = styles[1]
4. 一次运行可以生成5张图片,如果需要修改生成图片的数量,可以修改
num_generate = 55. 输出结果在genereted目录下 output_dir = './generated'
运行run_inference.py 等待图片生成