LoginSignup
1

More than 5 years have passed since last update.

github releaseを使って猫の線画を着色するモデルの配布を始めた

Last updated at Posted at 2017-08-09

概要

以前書いた記事で学習に使ったデータセットと訓練済みモデルの配布を、githubのリリース機能を使って配布しました。

URL

使い方

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

描画デモページを起動する

  • 以下の手順でサーバーを起動します
$ cd server
$ python serve.py --local_models_dir ../../export-cats
loading model edges2cats_AtoB
listening on :8000
  • export-catsディレクトリ以下にはedges2cats_AtoBディレクトリがあり、その中に訓練済みモデルがあるものとします
  • http://localhost:8000/ 等にブラウザでアクセスしてください

参考

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1