带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」( 二 )

带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」
文章插图
2.3.2 获取测试图片将待上色图片移动至data目录 。本样例使用图片方式获取如下
cd ../mkdir datacd datawget https://c7xcode.obs.cn-north-4.myhuaweicloud.com/models/colorization_picture-python/dog.png --no-check-certificate上述命令运行成功的截图如下:

带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
2.3.3 运行推理工程进入scripts目录,修改run.sh文件中INPUT_PIC变量为输入图片的路径,本示例为"…/data/dog.png" , 这个可以保持默认,暂时不做修改 。
此外,还要修改MX_SDK_HOME环境变量为SDK实际安装路径 。如果你不知道或不确定的话,我们下来查看一下,执行如下命令,即可查看当前的环境变量:
export执行后得到的结果截图如下,我们可以找到MX_SDK_HOME了,那么直接用来修改run.sh就好了 。
带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
修改后的run.sh截图:
带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
修改完成后,按照如下命令执行脚本:
cd ..# 回到Colorization目录下mkdir outcd ../scriptsbash run.sh这时 , 可能会报错,如下图所示,这是没安装OpenCV,其实前面的环境约束已经说了要安装OpenCV依赖的 , 我们来安装一下 , 执行如下命令:
pip3 install opencv-python --user-i https://pypi.tuna.tsinghua.edu.cn/simple之后再执行脚本,启动程序,就能运行成功了,部分运行截图如下所示:
带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图

带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
输出结果保存在out目录下,下载至本地查看图片上色是否合理,下面给大家展示一下前后效果:
带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
总的来说,除了狗的左侧耳朵的颜色有点问题外,效果还是很不错的,颜色比较自然,符合人类认知,感觉和拍摄得到的自然图片没什么区别哈 。
好了,到此这个案例基本就体验完成了,还是很不错的 。感兴趣的同学 , 可以将图片换成自己想要的图片来试试上色效果哦,注意修改图片名称和可能要修改的尺寸参数哦 。
2.4 基于MindStudio的运行2.4.1 启动MindStudio官方提供的镜像中有MindStudio的安装包,但是ECS默认是root用户登录,而在root用户下启动失败,应该是要HwHiAiUser才可以,但是HwHiAiUser用户在/root目录下无权限,无法启动存放在此的MindStudio,有点尴尬 。
所以,我要将MindStudio拷贝到HwHiAiUser的目录下,授予权限来使用,具体命令如下,注意以下命令是在root用户下操作:
cp /root/MindStudio_5.0.RC1_linux.tar.gz /home/HwHiAiUser/cd /home/HwHiAiUser/chown HwHiAiUser:HwHiAiUser MindStudio_5.0.RC1_linux.tar.gz切换到HwHiAiUser用户,再来执行:
tar -zxvf MindStudio_5.0.RC1_linux.tar.gz解压完成的截图如下所示:
带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
下面来启动MindStudio了,执行如下命令:
cd MindStudio/bin./MindStudio.sh初次启动可能有点慢,这也与当时网络状况有关,稍作等待即可,启动成功截图如下:
带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
还记得我们之前成功运行的黑白图像上色的工程吗?现在来打开看看 。
带你体验给黑白照片上色的感觉「带你体验给黑白照片上色」

文章插图
注意,MindStudio的图形界面和我们的操作都是通过ssh来连接和传输的,因此网络质量还是有较大影响的 , 如果网络状况好那就事半功倍了 。