AI自动P图：maskrcnn+inpaintting消除并修复指定区域图像

采用昇腾800-3000推理服务器实现基于maskrcnn和imageinpainting的图像区域消除和修复模型推理。mask-rcnn模型主要用于图像语义分割，语义分割后，image-inpaiting模型通过指定的坐标对需要消除的区域进行修复。

昇腾开源案例：

https://gitee.com/ascend/samples/tree/master/python/level3_multi_model/mask_rcnn_image_inpainting

另外，昇腾官网在线实验地址：

https://www.hiascend.com/zh/developer/mindx-sdk/imageInpainting

这里使用源码进行AI多模型图像区域消除推理功能的部署和实现。

一、测试环境

硬件(Atlas800-3000推理服务器)、操作系统(Ubuntu18.04、CANN5.0.3-alpha005)、开发语言(python3.7)

二、部署流程

1.根据要求配置好昇腾环境和依赖；

2.下载源码：git clone https://gitee.com/ascend/samples.git；

3.获取模型文件。

maskrcnn模型：

https://modelzoo-train-atc.obs.cn-north- 4.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/mask_rcnn/maskrcnn_mindspore.air

AIPP配置文件：

https://modelzoo-train-atc.obs.cn-north-4.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/mask_rcnn/aipp_rgb.cfg

单算子文件：

https://c7xcode.obs.myhuaweicloud.com/models/imageinpainting_hifill/matmul_27648.json

inpainting模型：

wget https://c7xcode.obs.myhuaweicloud.com/models/imageinpainting_hifill/hifill.om

4.模型转换。

转换maskrcnn模型：

atc --input_format=NCHW --framework=1 --model="maskrcnn_mindspore.air" --input_shape="x:1, 3, 768, 1280; im_info: 1, 4" --output="maskrcnn_mindspore_rgb" --insert_op_conf="aipp_rgb.cfg" --precision_mode=allow_fp32_to_fp16 --soc_version=Ascend310 --op_select_implmode=high_precision

转换单算子：

atc --singleop=matmul_27648.json --output= 0_BatchMatMul_0_0_1_1_1024_1024_0_0_1_1_1024_27648_0_0_1_1_1024_27648/ --soc_version=Ascend310

5.数据准备

下载测试图像：

https://modelzoo-train-atc.obs.cn-north-4.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/mask_rcnn/photo1.jpg

在源代码项目进入项目文件：

cd /sample/python/ level3_multi_model/ mask_rcnn_image_inpainting

新建几个目录：

mkdir data mask model output

将转换后的om模型文件放入model路径下，测试图像存入data文件夹。

6.执行测试

在mask_rcnn_image_inpainting路径下，终端输入命令：python3 src/ mask_rcnn.py 410 664。

执行成功后，分别进入mask和output文件夹查看结果。

效果：

AI自动P图：maskrcnn+inpaintting消除并修复指定区域图像相关推荐

R语言ggplot2可视化：使用ggpubr包的ggboxplot函数可视化分组箱图、使用bgcolor函数自定义指定可视化图像的背景色
R语言ggplot2可视化:使用ggpubr包的ggboxplot函数可视化分组箱图.使用bgcolor函数自定义指定可视化图像的背景色目录
AI智能生图技术的未来趋势
输入几个名词,就能得到一幅画,白丁秒变"画家":输入一张照片,得到一张二次元美照,而且还创意无限--今年以来,AI(人工智能)绘画火出了圈,大家享受着科技带来的免费作画的乐趣,甚至 ...
分析| AI智能生图前景几何?
如果说此前的AI绘画是人工智障,"AI不仁,以万物为刍狗",一不小心就把你"画"成了狗:那么现在,关于驯服AI的攻略贴则飘满了中国互联网,"大师之作& ...
python自动抠头像图_Python实现AI自动抠图实例解析
一.简介抠图是用PS? 用魔棒和快速选择工具? 遇到复杂背景怎么办? 最近发现一个神奇的工具--Remove Image Background 它是基于Python.Ruby和深度学习技术开发,通过 ...
实现一款俄罗斯方块小游戏非常简单！但是要实现AI自动俄罗斯方块才算牛逼！
前言最近刷抖音,看到一个玩俄罗斯方块的直播,居然玩到九万多分,也是个奇人,关键还有大几千人在那里看直播!这游戏有那么难吗?我试着玩了一下,最高也就玩到一千多分就玩不下去了!后面就图像想到,我是不是可 ...
Pytorch+LSTM+AI自动写诗实战
文章目录 1.数据集和任务定义 2.读取数据集 3.数据预处理 4.数据制作 5.定义网络结构: 6.测试网络 7.可视化 8.总结 1.数据集和任务定义本次采用的是唐诗数据集,一共有接近60000 ...
AI自动剪辑生成视频探索实践
点击上方"LiveVideoStack"关注我们 ▲扫描图中二维码或点击阅读原文▲了解音视频技术大会更多信息 // 编者按:随着短视频时代的到来,以短视频作为载体进行音乐推广具备 ...
python人工自动抠图_Python - AI自动抠图
一.简介抠图是用PS? 用魔棒和快速选择工具? 遇到复杂背景怎么办? 最近发现一个神奇的工具--Remove Image Background 它是基于Python.Ruby和深度学习技术开发,通过 ...
python人物抠图算法_Python实现AI自动抠图实例解析
一.简介抠图是用PS? 用魔棒和快速选择工具? 遇到复杂背景怎么办? 最近发现一个神奇的工具--Remove Image Background https://www.remove.bg/zh 它是 ...

AI自动P图：maskrcnn+inpaintting消除并修复指定区域图像

AI自动P图：maskrcnn+inpaintting消除并修复指定区域图像相关推荐

最新文章

热门文章