概要
以前書いた記事で学習に使ったデータセットと訓練済みモデルの配布を、githubのリリース機能を使って配布しました。
URL
- Release Publish pretrained model and dataset
- https://github.com/knok/pix2pix-tensorflow/releases/tag/v1.0
使い方
TensorFlow実装を使います。
自分で訓練をする
- データセットをダウンロード、展開する
- 以下を実行し訓練を行う
$ python pix2pix.py --mode train --input_dir ./joined \
--which_direction AtoB --output_dir trained-model --max_epochs 500
- 出来上がったモデルをexportする
$ python pix2pix.py --mode export \
--checkpoint trained-model \
--output_dir ../models/edges2cats_AtoB
- 事前訓練モデルを使う場合、export-cats-model.tar.gzを取得、展開してください
描画デモページを起動する
- 以下の手順でサーバーを起動します
$ cd server
$ python serve.py --local_models_dir ../../export-cats
loading model edges2cats_AtoB
listening on :8000
- export-catsディレクトリ以下にはedges2cats_AtoBディレクトリがあり、その中に訓練済みモデルがあるものとします
- http://localhost:8000/ 等にブラウザでアクセスしてください
参考
-
GitHubのリリース機能を使う - Qiita
- githubリリースを使うとCDNで配布してくれる点がありがたいです