昇腾社区首页
EN
注册
AnimeGan图像风格迁移

概述

本应用以昇腾Atlas 200I DK A2为主要的硬件平台,基于MindX SDK开发,可以实现从真实世界图片到动漫风格图片的风格转换,在昇腾芯片上对输入图片进行图像风格转换并将转换结果保存,支持多档次的动态分辨率输入。

功能模块介绍

本案例流程如下:

1.      输入类型可以是图片数据(jpg图片序列)

2.      通过OpenCV进行图像尺寸大小变换的预处理,以符合动态分辨率模型的档位。

3.      通过调用MindX SDK提供的图像解码插件mxpi_imagedecoder,解码后获取图像数据。

4.      通过调用MindX SDK提供的图像标准化插件mxpi_imagenormalize,将图像数据标准化到[-1,1]

5.      调用MindX_SDK的mxpi_tensorinfer插件,将尺寸变换后的图像数据输入AnimeGAN模型进行推理。

6.      风格转换后处理,使用自己编译的插件 animeganpostprocessor,进行图像风格转换的后处理,并保存文件。

支持说明

本应用同样支持Atlas200 DK,应用链接