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
画像ファイル
テスト用の画像も必要です。今回は以下の画像を用意しました。

実行
環境変数の設定をしていない場合は、コマンドラインで以下を実行します。
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
実行結果
コマンドを実行すると、左側に元々の画像にマスクが掛かった画像、右側にマスク部分を補完した画像が表示されます。だいたい戻ってますね。

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






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

フリーのITエンジニア(何でも屋さん)。趣味は渓流釣り、サッカー観戦、インラインホッケー、アイスホッケー、RaspberryPiを使った工作など。AI活用に興味があり試行錯誤中です。