其他分享
首页 > 其他分享> > 超酷! Atlas给黑白视频“上色”

超酷! Atlas给黑白视频“上色”

作者:互联网

摘要:随着人工智能技术发展,AI已经能够为黑白的老视频 “上色”,重现昔日的情景,让黑白图像变得栩栩如生。

怎么样,是不是看起来题目跟昨天的博客名称差不多?昨天是图片,今天是视频。从道理上说,将视频中的图像一一抓取,上色,合起来就是视频了。。。(我猜原理就是这样的)

所以,让我们把18.04的quick-start的demo折腾完吧。这次是根据 https://gitee.com/lovingascend/quick_start/blob/master/Atlas200DK_separate_MD/video_sample.md ,将视频给上色了。

闲话少说,我们按照说明来,使用”开发环境交叉编译第三方库,再将第三方库导入运行环境,提供运行调用“的方法。

先在虚拟机环境装presentagent:

然后依次照做:

bash configure

make -j8(编译的时间会有点长。。。)

sudo make install

下面编译arm版本的so,为迁移到开发板做准备:

make distclean

./configure --build=x86_64-linux-gnu --host=aarch64-linux-gnu --with-protoc=protoc

make -j8(耐心地等编译结束。。。)

sudo make install

下面到运行环境同步刚才编译好的第三方库:

注意这里手册有个bug:(scp需要加上一层arm目录),否则会找不到文件:scp $HOME/ascend_ddk/arm/lib/libpresenteragent.so HwHiAiUser@192.168.1.2:/home/HwHiAiUser

然后登录开发板,再拷贝so文件:

我们再做最后的一步,获取视频项目:

回到虚拟机的开发环境:

unzip colorization_video.zip 解压

建好目录,下载模型和权重文件:(好像跟上次是一样的)

回到MindStudio.关闭原来的工程并打开新下载的工程:

下面的步骤跟上次图片上色的那个基本是一致的:转换模型,加载模型到项目中,编译,运行。。。

个人感觉模型转换可以省略,因为图片上色的时候已经做过了。这次仅仅选择了加载。

检查下虚拟网卡的地址 192.168.1.223

以及 script/presentserver/display/config/config.conf中的presenter_server_ip 是否对应:

没问题。

再检查 src/colorize_process.cpp 106行地址是否跟前面对应:

也没问题。

修改编译参数(每次手册都说要该centos7.6,其实这倒是缺省的。倒是Target Architecture每次都要改成aarch64.手册每次都不说。只好笔者帮你说了。。。

点击Build。。。会生成build和out文件夹:

启动colorization-video应用的Presenter Server主程序:

居然报错了!!

看来这里还不能完全按照手册来,因为前面虽然都是按照手册装的,但是系统内分布了多个python版本,有apt-get安装的python3,也有用源码编译的python3.7.5.那么既然前面用的都是3.7.5.,那么这里的命令就应该改为:

python3.7.5 script/presenterserver/presenter_server.py --app=display & (手工加粗加红)

可以切换到 MindStudio Run一下了:先配置:

apply, ok后,再运行:

好像缺个动态库 libascendcl.so

经 @Jokey 大大的指点,

去开发板后台检查文件:

有这个文件,而且LD_LIBRARY_PATH也指向了这个so所在的目录:

另外,到开发板的工程out目录下,手工执行,好像是可以执行完毕的:

本来想修改下run.sh,在前面增加LD_LIBRARY_PATH变量。

结果发现,每次修改,都会被run程序将后台还原成一行的run.sh

这个就需要专家来回答了。暂时不管。。。(以后管。。。)

那我们看看在运行的时候,presenter_server怎么样?

在虚拟机打开浏览器:http://127.0.0.1:7009

可以看到有一路视频,点击view的名称,会弹出view的窗口,在run.sh执行的过程中,这里会有变化,截几个屏:

从视频效果来看,应该还是较为流畅的。fps=14.

我把原始视频和上色后的视频都传上来了。有兴趣的童鞋可以看下效果。剩下那个运行的问题,就等着专家来解决了。。。

另外,笔者去腾讯视频拍了一段 卓别林《摩登时代》的1分钟视频(参见附件),将其传入开发板的data目录:

然后run的参数再改成这个mp4的名称:

最后看到的效果如下:

转换好的视频(只录了一点点,因为太慢了影响我的感受,就没录完。。。)也上传到附件。

从转换的结果可以看出,fps=8.感觉有点慢。可能是200DK的推理能力还没这么快,也许300i可以快一点。。。

附上历史折腾清单:

华为Atlas 200DK初体验——写在昇腾新品发布的第二天  https://bbs.huaweicloud.com/blogs/193847

Atlas 200DK系统折腾纪实:(1)论图片及18.04.1的诞生  https://bbs.huaweicloud.com/blogs/194291

Atlas 200DK系统折腾纪实:(2)视频的诞生 https://bbs.huaweicloud.com/blogs/194641

Atlas 200DK系统折腾纪实:(3)强大的yolo3物体检测——你想看的内容已被系统提取 https://bbs.huaweicloud.com/blogs/194816

Atlas 200DK系统折腾纪实:(4)18.04.1的软件安装及双系统切换验证 https://bbs.huaweicloud.com/blogs/195227

Atlas 200DK系统折腾纪实:(5)Atlas让黑白照片原彩重现 https://bbs.huaweicloud.com/blogs/195391

(全文完,谢谢阅读,本文中的问题看看下次能不能解决。。)

 black-white_video.rar 1.02MB 

 out_modengshidai.rar 2.26MB 

 modengshidai.rar 9.13MB 

 上色后的视频0826.rar 2.29MB 

 

点击关注,第一时间了解华为云新鲜技术~

标签:视频,200DK,黑白,编译,https,Atlas,超酷,com
来源: https://blog.51cto.com/u_15214399/2824527