【Gradio应用】图片修复工具

★★★ 本文源自AlStudio社区精品项目，【点击此处】查看更多精品内容 >>>

API 使用示例

下面演示代码中，推理函数 predict() 的输入/输出形状为 [C:H:W]，其中C通道的顺序为BGR。因此，输出的图片需转为RGB模式。输入/输出的真实分辨率为 512 x 512，其他尺寸的图片会经过 resize() 处理。输入的图片必须为PNG格式，JPG等其他压缩格式的图片需转换为PNG格式，再进行擦除操作。否则，会导致擦除区域标识错误。

from g import predict
import cv2
import matplotlib.pyplot as plt
%matplotlib inline

img = cv2.imread('examples/1.png')
output = predict(img)
output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
plt.imshow(output)
plt.show()

本应用基于AOT-GAN结构搭建，并进行了改进优化，模型指标大幅提升并获“兴智杯”大赛技术模型赛决赛一等奖。模型复现请参考项目：『2022信通院兴智杯：论文复现赛』AOT-GAN 论文复现

此文章为搬运
原项目链接