AOT-GAN + onnx部署 + gradio 实现在线图像修复

发布时间：2024年01月14日

运行环境说明，其依赖： 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目录下，该模型为人脸编辑模型。在这里插入图片描述

文章来源:https://blog.csdn.net/a486259/article/details/135563289
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
最新文章