tinghuiz / SfMLearner

An unsupervised learning framework for depth and ego-motion estimation from monocular videos
MIT License
1.97k stars 557 forks source link

train.py InvalidArgumentError: Shape mismatch in tuple component 0. Expected [128,416,6], got [128,416,2] #112

Open gth901007 opened 5 years ago

gth901007 commented 5 years ago

Traceback (most recent call last): File "train.py", line 46, in tf.app.run() File "/home/zyw/anaconda3/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 125, in run _sys.exit(main(argv)) File "train.py", line 43, in main sfm.train(FLAGS) File "/home/zyw/project/Sfmlearner/SfMLearner-master/SfMLearner.py", line 262, in train self.save(sess, opt.checkpoint_dir, gs) File "/home/zyw/anaconda3/lib/python3.6/contextlib.py", line 99, in exit self.gen.throw(type, value, traceback) File "/home/zyw/anaconda3/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 1005, in managed_session self.stop(close_summary_writer=close_summary_writer) File "/home/zyw/anaconda3/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 833, in stop ignore_live_threads=ignore_live_threads) File "/home/zyw/anaconda3/lib/python3.6/site-packages/tensorflow/python/training/coordinator.py", line 389, in join six.reraise(*self._exc_info_to_raise) File "/home/zyw/anaconda3/lib/python3.6/site-packages/six.py", line 693, in reraise raise value File "/home/zyw/anaconda3/lib/python3.6/site-packages/tensorflow/python/training/queue_runner_impl.py", line 252, in _run enqueue_callable() File "/home/zyw/anaconda3/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1205, in _single_operation_run self._call_tf_sessionrun(None, {}, [], target_list, None) File "/home/zyw/anaconda3/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1350, in _call_tf_sessionrun run_metadata) tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape mismatch in tuple component 0. Expected [128,416,6], got [128,416,2] How to solve this problem? Thank you!

KerunLi commented 3 years ago

hello, I have the same problem, have you solve it?