colorization_demoの実行(macOS編)

はじめに

Open Model Zoo内のDemoに格納されている、colorization_demo を使ってみましょう。

環境

今回はmacOSで実行してみます。(もちろん他OSでも同等です)

MacBook Pro (13-inch, 2018, Four Thunderbolt 3 Ports)
2.7 GHz クアッドコアIntel Core i7 メモリ16 GB
macOS Big Sur 11.1
Python 3.7.7
openvino 2021.2.185

モデルの確認

models.lstを開いて、使用するモデルを確認します。colorization-v2、colorization-siggraphの2つが利用出来る事がわかります。モデル未入手の場合は、モデルダウンローダーを使って入手してください。

# This file can be used with the --list option of the model downloader.
colorization-v2
colorization-siggraph

ヘルプの確認

% python3 colorization_demo.py -h
usage: colorization_demo.py [-h] -m MODEL [-d DEVICE] -i "<path>" [--no_show]
                            [-v] [-u UTILIZATION_MONITORS]

Options:
  -h, --help            Help with the script.
  -m MODEL, --model MODEL
                        Required. Path to .xml file with pre-trained model.
  -d DEVICE, --device DEVICE
                        Optional. Specify target device for infer: CPU, GPU,
                        FPGA, HDDL or MYRIAD. Default: CPU
  -i "<path>", --input "<path>"
                        Required. Input to process.
  --no_show             Optional. Disable display of results on screen.
  -v, --verbose         Optional. Enable display of processing logs on screen.
  -u UTILIZATION_MONITORS, --utilization_monitors UTILIZATION_MONITORS
                        Optional. List of monitors to show initially.

実行してみます

チャップリンの動画を使いました。モデル毎に確認してみます。

% python3 colorization_demo.py
 -m /public/colorization-v2/FP16/colorization-v2.xml
 -i Charlie_Chaplin_Mabels_Strange_Predicament.avi 
% python3 colorization_demo.py
 -m /public/colorization-siggraph/FP16/colorization-siggraph.xml 
 -i Charlie_Chaplin_Mabels_Strange_Predicament.avi

動画でも確認してみましょう

colorization-v2

colorization-siggraph