閉じる

TensorFlowの練習 ~ Cifar10の画像をダウンロードしCNNしてみた ~

f:id:Vastee:20180826205452p:plain

WindowsにTensorflowをインストールしたので,試しにCNNしてみた.

なお,本記事の手順は以下の記事を参考にした.

blogs.yahoo.co.jp

必要スクリプトのダウンロード

以下のリポジトリから,

github.com

  • cifar10.py
  • cifar10_input.py
  • cifar10_train.py
  • cifar10_eval.py

をダウンロードし,同一ディレクトリに配置.

コードの書き換え

学習を早く終わらせるため,cifar10_train.pyを書き換える.

# 修正前
tf.app.flags.DEFINE_integer('max_steps', 100000,
"""Number of batches to run.""")
# 修正後
tf.app.flags.DEFINE_integer('max_steps', 2000,
"""Number of batches to run.""")

cifar10_train.py の実行

python cifar10_train.py # Lossが減少してゆく様子を眺めよう.

Tensorboard で結果をみてみよう

tensorboard --logdir=C:\tmp\cifar10_train # 自分の場合,EドライブにAnacondaをインストールしてしまった為,C:の部分をE:に変更した.

実行後,URLが出現するので,ブラウザでアクセスするとTensorboardを開くことができる.

Pocket

コメントを残す

あなたのメールアドレスは公開されません。必須項目には印がついています *

© 2020 CV・NLP TIPS | WordPress Theme: Annina Free by CrestaProject.