超酷! 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

 

(全文完,謝謝閱讀,本文中的問題看看下次能不能解決。。)

 

點擊關注,第一時間了解華為雲新鮮技術~


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM