本文主要是介绍AOT-GAN + onnx部署 + gradio 实现在线图像修复,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
运行环境说明,其依赖: 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目录下,该模型为人脸编辑模型。
这篇关于AOT-GAN + onnx部署 + gradio 实现在线图像修复的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!