質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

90.84%

  • Python

    6411questions

    Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

  • TensorFlow

    560questions

TensorFlowのsaver.restoreについて

解決済

回答 1

投稿

  • 評価
  • クリップ 0
  • VIEW 507

TakakiKuwabara

score 12

Python3のTensorFlowにて、下記のコードによりエラーが発生しました。
エラーの原因はtf.train.Saverにあるようなのですが、調べたところTensorFlowのバージョンが悪いのやら、
cifar10.pyの一部を書き換えが必要なのやら、何が正しいのかわからない状態です。

どなたかわかる方がいらっしゃれば、お教えいただけないでしょうか。

import sys
import tensorflow as tf
import cifar10

FLAGS = tf.app.flags.FLAGS
cifar10.NUM_CLASSES = 7
tf.app.flags.DEFINE_string('checkpoint_dir', '/tmp/cifar10_train',
                           """Directory where to read model checkpoints.""")

def evaluate(filename):
  with tf.Graph().as_default() as g:
    file   = tf.read_file(filename)
    image  = tf.image.decode_jpeg(file, channels = 3)
    image  = tf.image.resize_images(image, [32, 32])
    # cifar10は内部処理で32×3224×24に切り出して利用している
    image  = tf.image.resize_image_with_crop_or_pad(image, 24, 24)
    logits = cifar10.inference([image])

    # ここのkの値もクラス数と一致させる
    top_k_op = tf.nn.top_k(logits, k=7)

    variable_averages = tf.train.ExponentialMovingAverage(cifar10.MOVING_AVERAGE_DECAY)
    variables_to_restore = variable_averages.variables_to_restore()
    saver = tf.train.Saver(variables_to_restore)

    sess = tf.InteractiveSession()
    sess.run(tf.initialize_all_variables())
    ckpt = tf.train.get_checkpoint_state(FLAGS.checkpoint_dir)

    if ckpt and ckpt.model_checkpoint_path:
      saver.restore(sess, ckpt.model_checkpoint_path)

    else:
      print('No checkpoint file found')
      return

    tf.train.start_queue_runners(sess=sess)
    values, indices = sess.run(top_k_op)
    ratio = sess.run(tf.nn.softmax(values[0]))

    # 予想したラベルとそれぞれに対する確信度
    print(indices[0])
    print(ratio)

def main(argv=None):
  evaluate(sys.argv[1])

if __name__ == '__main__':
  tf.app.run()

エラー内容は下記の通りです。

Traceback (most recent call last):
  File "cifar10_sample.py", line 60, in <module>
    tf.app.run()
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 44, in run
    _sys.exit(main(_sys.argv[:1] + flags_passthrough))
  File "cifar10_sample.py", line 56, in main
    evaluate(sys.argv[1])
  File "cifar10_sample.py", line 40, in evaluate
    saver.restore(sess, ckpt.model_checkpoint_path)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 1439, in restore
    {self.saver_def.filename_tensor_name: save_path})
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 767, in run
    run_metadata_ptr)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 965, in _run
    feed_dict_string, options, run_metadata)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1015, in _do_run
    target_list, options, run_metadata)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1035, in _do_call
    raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [192,7] rhs shape= [192,10]
     [[Node: save/Assign_9 = Assign[T=DT_FLOAT, _class=["loc:@softmax_linear/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/cpu:0"](softmax_linear/weights, save/RestoreV2_9)]]

Caused by op 'save/Assign_9', defined at:
  File "cifar10_sample.py", line 60, in <module>
    tf.app.run()
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 44, in run
    _sys.exit(main(_sys.argv[:1] + flags_passthrough))
  File "cifar10_sample.py", line 56, in main
    evaluate(sys.argv[1])
  File "cifar10_sample.py", line 33, in evaluate
    saver = tf.train.Saver(variables_to_restore)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 1051, in __init__
    self.build()
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 1081, in build
    restore_sequentially=self._restore_sequentially)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 675, in build
    restore_sequentially, reshape)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 414, in _AddRestoreOps
    assign_ops.append(saveable.restore(tensors, shapes))
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 155, in restore
    self.op.get_shape().is_fully_defined())
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/ops/gen_state_ops.py", line 47, in assign
    use_locking=use_locking, name=name)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/framework/op_def_library.py", line 763, in apply_op
    op_def=op_def)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 2395, in create_op
    original_op=self._default_original_op, op_def=op_def)
  File "/Users/k_aki86/.pyenv/versions/3.5.1/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 1264, in __init__
    self._traceback = _extract_stack()

InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [192,7] rhs shape= [192,10]
     [[Node: save/Assign_9 = Assign[T=DT_FLOAT, _class=["loc:@softmax_linear/weights"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/cpu:0"](softmax_linear/weights, save/RestoreV2_9)]]
  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 1

check解決した方法

0

下記の情報に従えば解決しました。
https://groups.google.com/a/tensorflow.org/forum/#!topic/discuss/4xjc7tSrb18

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 90.84%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る

  • Python

    6411questions

    Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

  • TensorFlow

    560questions