Hi,
Thank you for this implementation of UNET.
I am trying to use your code and replicate your results. However I am getting the following error.
I was wondering if you can help.
`During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/home/potis/Desktop/Pycharm/unettf/main.py", line 30, in
tf.app.run()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 125, in run
_sys.exit(main(argv))
File "/home/potis/Desktop/Pycharm/unettf/main.py", line 23, in main
getattr(model, args.option)()
File "/home/potis/Desktop/Pycharm/unettf/model.py", line 143, in train
classifier.train(input_fn=input_fn_train, hooks=[logging_hook])
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 366, in train
loss = self._train_model(input_fn, hooks, saving_listeners)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 1119, in _train_model
return self._train_model_default(input_fn, hooks, saving_listeners)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 1135, in _train_model_default
saving_listeners)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 1333, in _train_with_estimator_spec
log_step_count_steps=self._config.log_step_count_steps) as mon_sess:
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 415, in MonitoredTrainingSession
stop_grace_period_secs=stop_grace_period_secs)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 826, in init
stop_grace_period_secs=stop_grace_period_secs)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 549, in init
self._sess = _RecoverableSession(self._coordinated_creator)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 1012, in init
_WrappedSession.init(self, self._create_session())
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 1017, in _create_session
return self._sess_creator.create_session()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 706, in create_session
self.tf_sess = self._session_creator.create_session()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 477, in create_session
init_fn=self._scaffold.init_fn)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/session_manager.py", line 281, in prepare_session
config=config)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/session_manager.py", line 211, in _restore_checkpoint
saver.restore(sess, ckpt.model_checkpoint_path)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 1752, in restore
{self.saver_def.filename_tensor_name: save_path})
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 900, in run
run_metadata_ptr)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1135, in _run
feed_dict_tensor, options, run_metadata)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1316, in _do_run
run_metadata)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1335, 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= [3,3,3,2,32] rhs shape= [3,3,3,1,32]
[[Node: save/Assign_106 = Assign[T=DT_FLOAT, _class=["loc:@conv3d/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](conv3d/kernel, save/RestoreV2:106)]]
Caused by op 'save/Assign_106', defined at:
File "/home/potis/Desktop/Pycharm/unettf/main.py", line 30, in
tf.app.run()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 125, in run
_sys.exit(main(argv))
File "/home/potis/Desktop/Pycharm/unettf/main.py", line 23, in main
getattr(model, args.option)()
File "/home/potis/Desktop/Pycharm/unettf/model.py", line 143, in train
classifier.train(input_fn=input_fn_train, hooks=[logging_hook])
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 366, in train
loss = self._train_model(input_fn, hooks, saving_listeners)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 1119, in _train_model
return self._train_model_default(input_fn, hooks, saving_listeners)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 1135, in _train_model_default
saving_listeners)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/estimator/estimator.py", line 1333, in _train_with_estimator_spec
log_step_count_steps=self._config.log_step_count_steps) as mon_sess:
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 415, in MonitoredTrainingSession
stop_grace_period_secs=stop_grace_period_secs)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 826, in init
stop_grace_period_secs=stop_grace_period_secs)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 549, in init
self._sess = _RecoverableSession(self._coordinated_creator)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 1012, in init
_WrappedSession.init(self, self._create_session())
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 1017, in _create_session
return self._sess_creator.create_session()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 706, in create_session
self.tf_sess = self._session_creator.create_session()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 468, in create_session
self._scaffold.finalize()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/monitored_session.py", line 214, in finalize
self._saver.build()
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 1296, in build
self._build(self._filename, build_save=True, build_restore=True)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 1333, in _build
build_save=build_save, build_restore=build_restore)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 775, in _build_internal
restore_sequentially, reshape)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 453, in _AddShardedRestoreOps
name="restore_shard"))
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 422, in _AddRestoreOps
assign_ops.append(saveable.restore(saveable_tensors, shapes))
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 113, in restore
self.op.get_shape().is_fully_defined())
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/ops/state_ops.py", line 219, in assign
validate_shape=validate_shape)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/ops/gen_state_ops.py", line 60, in assign
use_locking=use_locking, name=name)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3414, in create_op
op_def=op_def)
File "/home/potis/Desktop/Python_projects/VirtualEnv/skynet3/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 1740, in init
self._traceback = self._graph._extract_stack() # pylint: disable=protected-access
InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [3,3,3,2,32] rhs shape= [3,3,3,1,32]
[[Node: save/Assign_106 = Assign[T=DT_FLOAT, _class=["loc:@conv3d/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](conv3d/kernel, save/RestoreV2:106)]]
Process finished with exit code 1
`