トップページ -> 実践知識 -> TensorFlow のインストールおよびその応用例 -> pix2pix を動かしてみる
[サイトマップへ], サイト内検索:

pix2pix を動かしてみる

サイト構成 連絡先,業績など 実践知識 データの扱い コンピュータ 教材 サポートページ

公開されたソフトウエア,データを使い,Windows で pix2pix を動かしてみます. Python, TensorFlow を使います. 先人に感謝.

参考Webページ: affinelayer/pix2pix-tensorflow のページ, https://github.com/affinelayer/pix2pix-tensorflow

このWebページでは,pix2pix の Python 版が公開されているとともに,次のような 5種類のデータセットが公開されている.


前準備

Anaconda, git のインストール

前準備として,Python 開発環境のAnaconda, git のインストールが終わっていること.

TensorFlow のインストール

Windows での 手順は、 「Windows で TensorFlow のインストール」の Web ページで説明している.

GPU が使える場合には、「この Web ページの説明通りに「conda install -y tensorflow-gpu」を終えておくとよいでしょう。

以下,Windows での Anaconda と git をインストール済み, Tensorflow をインストール済みであるものとして説明を続ける.


pix2pixのインストール

謝辞:https://github.com/affinelayer/pix2pix-tensorflow で公開されている著作物を利用している

  1. Windows で,コマンドプロンプトを管理者として実行

  2. pix2pixのインストール

    mkdir c:\pytools
    cd c:\pytools
    rmdir /s /q pix2pix
    

    git clone https://github.com/affinelayer/pix2pix-tensorflow
    cd pix2pix-tensorflow
    

  3. データのダウンロード操作

    affinelayer/pix2pix-tensorflow の Web ページ に記載の通り行う.

    cd c:\pytools\pix2pix-tensorflow
    python tools\download-dataset.py facades
    

  4. facades の下の train と test にたくさんの画像ファイルがダウンロードされたことを確認する.

  5. モデルのトレーニングを行う操作

    ※ これも,affinelayer/pix2pix-tensorflow の Web ページ に記載の通り行う.

    # train the model (this may take 1-8 hours depending on GPU, on CPU you will be waiting for a bit)
    python pix2pix.py --mode train --output_dir facades_train --max_epochs 100 --input_dir facades\train  --which_direction BtoA
    

    BtoA について

    「--which_direction BtoA」と指定したので, 「右側」のような画像をコンピュータに与えたら,「左側」のような画像が生成されるように,モデルのトレーニングを行いなさいという指示になる.

  6. 終了の確認

    ※ GPU を使わない場合は,終了まで,かなり待つ.終了したらエラーメッセージが出ていないことを確認する.

  7. 学習結果のファイルの確認

    次に,「学習結果の出力先 (--output_dir オプション)」として指定した facades_train に,ファイルができていることを確認する.

  8. 確認のため,学習結果を tensorborard で表示

    次の手順で表示してみる

    1. tensorboard の起動

      ラスベリーパイ(Raspberry Pi)と Windows パソコンをネットワーク接続.

      このとき,起動メッセージに,URLが「http://hoge:6006」のように表示されるので確認する.

      実行手順例

    2. Web ブラウザで、先ほどの URL を開く

    3. GRAPHS」をクリックして確認する

  9. モデルのテストを行う操作

    画像を与えて,新しい画像を作ることで,学習の結果できた「モデル」のテストを行う.

    ※ これも,affinelayer/pix2pix-tensorflow の Web ページ に記載の通り行う.

    python pix2pix.py  --mode test --output_dir facades_test --input_dir facades\val  --checkpoint facades_train
    

    モデルのテストのプログラムは,それほど時間がかからない.終わったら,エラーメッセージが出ていないことを確認する.

    上のコマンドを実行すると,出力先として指定した「facades_test」に ファイルができるので確認する.

    ここにある index.html ファイルを,Web ブラウザで開くと,簡単に結果が確認できる.