進行推理時,需要編輯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 等待圖片生成