本文分享自华为云社区《》,作者:Tianyi_Li 。
前言在智能手机越来越普及的今天,拍摄一张色彩鲜艳、清晰的照片轻而易举。但是老照片没有如此“幸运”,大多为黑白。借助人工智能技术,可以一定程度上帮助老照片还原原来色彩。还记得之前火热的“老北京视频上色”吗?就是采用类似的技术啊。
这次介绍的是黑白图像上色应用,旨在华为自研的Ascend 310推理芯片上实现输入黑白图像,自动对黑白图像进行上色,还原彩色图像。
该应用案例整体流程如下图所示:
简单来说分为以下部分:
- 图像前处理:将输入的黑白图像通过OpenCV转换为Lab图,并抽取其中的L通道进行前处理;
- Colorization模型推理:将处理后的L通道传入模型进行推理,获得预测结果的a,b通道;
- 模型后处理:拼接原始L通道和预测得的a,b通道获得新的Lab图。
操作流程AI加速型 | ai1s.large.4 | 2vCPUs | 8GiB
MindStudio5.0.RC1_CANN5.1.RC1
Ubuntu 18.04 server 64bit
假设我们已经收到代金券,并通过官方共享的镜像购买了ECS。
1. 登录环境我使用的是MobaXterm,登陆界面如图所示:
2. 获取代码2.1 切换到普通用户默认登录是root用户,权限太大了,开发不太安全,我们先切换到普通用户,命令如下:
su - HwHiAiUser
切换成功后,终端的提示符会变成$,我的当前目录如下图所示:
2.2 获取代码我使用的是官方案例,执行如下命令下载:
git clone https://gitee.com/ascend/mindxsdk-referenceapps.git
运行成功截图如下: