【Gradio应用】图片修复工具
图片修复工具基于 Image Inpainting 模型,用于擦除图像中不想保留的部分,并自动补全背景。
·
★★★ 本文源自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 论文复现
此文章为搬运
原项目链接
更多推荐
已为社区贡献1438条内容
所有评论(0)