tensorflow运行SegNet问题
使用下面的segnet实现代码
https://github.com/tkuanlun350/Tensorflow-SegNet
在main中已经修改了camvid数据集里面三个txt文件的数据路径,但是报错
The model is set to Training
Max training Iteration: 20000
Initial lr: 0.001000
CamVid Image dir: /home/jiang/lane/SegNet/CamVid/train.txt
CamVid Val dir: /home/jiang/lane/SegNet/CamVid/val.txt
Batch Size: 1
Log dir: /tmp3/first350/TensorFlow/Logs
Filling queue with 146 CamVid images before starting to train. This will take a few minutes.
Filling queue with 146 CamVid images before starting to train. This will take a few minutes.
2017-11-06 23:39:54.802996: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
Traceback (most recent call last):
File "/home/jiang/lane/Tensorflow-SegNet-master/main.py", line 52, in <module>
tf.app.run()
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 48, in run
_sys.exit(main(_sys.argv[:1] + flags_passthrough))
File "/home/jiang/lane/Tensorflow-SegNet-master/main.py", line 49, in main
model.training(FLAGS, is_finetune=False)
File "/home/jiang/lane/Tensorflow-SegNet-master/model.py", line 389, in training
sess.run(init)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 889, in run
run_metadata_ptr)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1120, in _run
feed_dict_tensor, options, run_metadata)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1317, in _do_run
options, run_metadata)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1336, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels:
<no registered kernels>
[[Node: pool1 = MaxPoolWithArgmax[T=DT_FLOAT, Targmax=DT_INT64, ksize=[1, 2, 2, 1], padding="SAME", strides=[1, 2, 2, 1]](conv1/Relu)]]
Caused by op u'pool1', defined at:
File "/home/jiang/lane/Tensorflow-SegNet-master/main.py", line 52, in <module>
tf.app.run()
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 48, in run
_sys.exit(main(_sys.argv[:1] + flags_passthrough))
File "/home/jiang/lane/Tensorflow-SegNet-master/main.py", line 49, in main
model.training(FLAGS, is_finetune=False)
File "/home/jiang/lane/Tensorflow-SegNet-master/model.py", line 372, in training
loss, eval_prediction = inference(train_data_node, train_labels_node, batch_size, phase_train)
File "/home/jiang/lane/Tensorflow-SegNet-master/model.py", line 182, in inference
padding='SAME', name='pool1')
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/gen_nn_ops.py", line 3525, in max_pool_with_argmax
padding=padding, Targmax=Targmax, name=name)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 2956, in create_op
op_def=op_def)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 1470, in __init__
self._traceback = self._graph._extract_stack() # pylint: disable=protected-access
InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels:
<no registered kernels>
[[Node: pool1 = MaxPoolWithArgmax[T=DT_FLOAT, Targmax=DT_INT64, ksize=[1, 2, 2, 1], padding="SAME", strides=[1, 2, 2, 1]](conv1/Relu)]]
不知道怎么回事 查半天也没查出来