Image Inpainting Demoの実行(Windows編)

OpenVINO 2020.2に入っているimage inpainting demoを実行してみましょう。まずは概要の動画をご確認ください。

デモのフォルダは以下になります。

C:\Program Files (x86)\IntelSWTools\openvino\inference_engine\demos\python_demos\image_inpainting_demo

実行の前に

サンプルを実行する前に、READMEファイルを確認しておきましょう。オプションスイッチや実行例が記載されています。

モデル

今回は以下のモデルを利用しました。(image_inpainting_demoフォルダにある、models.txtで利用するモデルがわかります。)

  • gmcnn-places2-tf

モデルをダウンロードします。モデルの保存先(-o)は適時読み替えてください。

cd C:\Program Files (x86)\IntelSWTools\openvino\inference_engine\demos\python_demos\image_inpainting_demo

python downloader.py --list "C:\Program Files (x86)\IntelSWTools\openvino\inference_engine\demos\python_demos\image_inpainting_demo\models.lst" -o C:\Users\ユーザ名\Documents\Intel\OpenVINO\openvino_models

コマンドを実行すると、ダウンロードが始まります。

################|| Downloading models ||################ ========== Downloading C:\Users\ユーザ名\Documents\Intel\OpenVINO\openvino_models\public\gmcnn-places2-tf\frozen_model.pb ... 100%, 48608 KB, 3511 KB/s, 13 seconds passed ################|| Post-processing ||################

ダウンロードしたモデルを中間形式に変換します。モデルの読み込み先(-d)と保存先(-o)は適時読み替えてください。

python converter.py --list "C:\Program Files (x86)\IntelSWTools\openvino\inference_engine\demos\python_demos\image_inpainting_demo\models.lst" -d C:\Users\ユーザ名\Documents\Intel\OpenVINO\openvino_models -o C:\Users\ユーザ名\Documents\Intel\OpenVINO\openvino_models

[ SUCCESS ]が表示されて、モデルが変換されたら完了です。

画像ファイル

テスト用の画像も必要です。今回は以下の画像を用意しました。

SUZUKI BANDIT 250V

実行

環境変数の設定をしていない場合は、コマンドラインで以下を実行します。

C:¥Program Files (x86)¥IntelSWTools¥openvino¥bin¥setupvars.bat

以下のコマンドでImage Inpainting Demoを実行します。
モデルの格納先、画像の保存先フォルダは適時読み替えてください。

python image_inpainting_demo.py -m C:\Users\ユーザ名\Documents\Intel\OpenVINO\openvino_models\public\gmcnn-places2-tf\FP16\gmcnn-places2-tf.xml -i C:\Users\ユーザ名\Desktop\画像.jpg

実行結果

コマンドを実行すると、左側に元々の画像にマスクが掛かった画像、右側にマスク部分を補完した画像が表示されます。だいたい戻ってますね。

左側:マスクが掛かっている画像 ー> 右側:補完した画像

他の画像でも試してみましょう。

毎度おなじみの、ふーちゃん
まあまあ補完されていますね。少し離れれば分からないレベル。
近くの川で釣って来たヤマメ3匹
結構補完されていますね。
京都醸造さんのクラフトビール。美味しいです。
ビンの形は補完されてますね。ラベルは難しいでしょうね。

image inpainting demoの実行、如何だったでしょうか。いろいろと画像を試してみてくださいね。