运行环境说明,其依赖: torch + python-opencv + onnxruntime + gradio + numpy + einops
1、模型转onnx
可以直接下载博主导出的onnx模型 https://download.csdn.net/download/a486259/88736610
1.1 预训练pt模型下载
下载place2 预训练模型,图中红色部分,模型保存为place2.pt ,模型存储到AOT-GAN-for-Inpainting-master\src目录下,该模型为通用图像修复模型。下载 CELEBA-HQ 预训练模型,图中红色部分,模型保存为face.pt,模型存储到AOT-GAN-for-Inpainting-master\src目录下,该模型为人脸编辑模型。