search

twke18 / Adaptive_Affinity_Fields

Adaptive Affinity Fields for Semantic Segmentation
https://jyhjinghwang.github.io/projects/aaf.html
Other
262 stars 62 forks source link

wrong in inference.py #11

Closed Jee-King closed 5 years ago

Jee-King commented 6 years ago

Hi， Could you provide your training model on voc 2012? I didn't find him. Thank you very much!

Jee-King commented 6 years ago

I got this error

`Caused by op u'save/RestoreV2_541', defined at: File "pyscripts/inference/inference.py", line 226, in main() File "pyscripts/inference/inference.py", line 136, in main loader = tf.train.Saver(var_list=restore_var) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 1218, in init self.build() File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 1227, in build self._build(self._filename, build_save=True, build_restore=True) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 1263, in _build build_save=build_save, build_restore=build_restore) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 751, in _build_internal restore_sequentially, reshape) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 427, in _AddRestoreOps tensors = self.restore_op(filename_tensor, saveable, preferred_shard) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 267, in restore_op [spec.tensor.dtype])[0]) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/gen_io_ops.py", line 1021, in restore_v2 shape_and_slices=shape_and_slices, dtypes=dtypes, 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

NotFoundError (see above for traceback): Tensor name "resnet_v1_101/block5/pool6/conv1/weights" not found in checkpoint files /home/wyr/zjq/Adaptive_Affinity_Fields-master/resnet_v1_101.ckpt [[Node: save/RestoreV2_541 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2_541/tensor_names, save/RestoreV2_541/shape_and_slices)]] [[Node: save/RestoreV2_170/_501 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name="edge_1598_save/RestoreV2_170", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"]()]] `

twke18 commented 6 years ago

Hi, here is the link for trained model on VOC and Cityscapes.

Jee-King commented 5 years ago

Thank you very much!!