TensorFlow-ENet
TensorFlow-ENet copied to clipboard
ScaterNd Error in train_enet.py file
I am getting the following error on running the train_enet.py file in python=3.6 TensorFlow=1.12. Can someone help with this error?
INFO:tensorflow:Error reported to Coordinator: <class 'tensorflow.python.framework.errors_impl.InvalidArgumentError'>, indices[172800] = [1, 91, 1, 0] does not index into shape [25,90,120,64]
[[node ENet_1/unpool/ScatterNd (defined at /media/rohit/Data/water-segmentation/TensorFlow-ENet/enet.py:112) = ScatterNd[T=DT_FLOAT, Tindices=DT_INT32, _device="/job:localhost/replica:0/task:0/device:CPU:0"](ENet_1/unpool/transpose, ENet_1/unpool/Reshape_2, ENet_1/unpool/ScatterNd/shape)]]
Caused by op 'ENet_1/unpool/ScatterNd', defined at:
File "train_enet.py", line 337, in <module>
run()
File "train_enet.py", line 234, in run
skip_connections=skip_connections)
File "/media/rohit/Data/water-segmentation/TensorFlow-ENet/enet.py", line 467, in ENet
pooling_indices=pooling_indices_2, output_shape=inputs_shape_2, scope=bottleneck_scope_name+'_0')
File "/home/rohit/anaconda3/envs/tf-gpu/lib/python3.6/site-packages/tensorflow/contrib/framework/python/ops/arg_scope.py", line 182, in func_with_args
return func(*args, **current_args)
File "/media/rohit/Data/water-segmentation/TensorFlow-ENet/enet.py", line 324, in bottleneck
net_unpool = unpool(net_unpool, pooling_indices, output_shape=output_shape, scope='unpool')
File "/media/rohit/Data/water-segmentation/TensorFlow-ENet/enet.py", line 112, in unpool
ret = tf.scatter_nd(indices, values, output_shape)
File "/home/rohit/anaconda3/envs/tf-gpu/lib/python3.6/site-packages/tensorflow/python/ops/gen_array_ops.py", line 7077, in scatter_nd
"ScatterNd", indices=indices, updates=updates, shape=shape, name=name)
File "/home/rohit/anaconda3/envs/tf-gpu/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "/home/rohit/anaconda3/envs/tf-gpu/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 488, in new_func
return func(*args, **kwargs)
File "/home/rohit/anaconda3/envs/tf-gpu/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3274, in create_op
op_def=op_def)
File "/home/rohit/anaconda3/envs/tf-gpu/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1770, in __init__
self._traceback = tf_stack.extract_stack()