Как я могу решить эту проблему, возникшую при прогнозировании выходных данных с использованием модели Deeplab? - PullRequest
Новая
       15

Как я могу решить эту проблему, возникшую при прогнозировании выходных данных с использованием модели Deeplab?

0 голосов
/ 21 октября 2018

Я использую свой собственный набор данных для выполнения задачи сегментации.После обучения я запускаю vis.py, и он возвращает ошибку следующим образом: 

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/public/Desktop/DeepLabV3+/models-master/research/deeplab/vis.py", line 321, in <module>
    tf.app.run()
  File "/home/public/anaconda3/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 125, in run
    _sys.exit(main(argv))
  File "/home/public/Desktop/DeepLabV3+/models-master/research/deeplab/vis.py", line 307, in main
    image_id_offset += FLAGS.vis_batch_size
  File "/home/public/anaconda3/lib/python3.6/contextlib.py", line 99, in __exit__
    self.gen.throw(type, value, traceback)
  File "/home/public/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/public/anaconda3/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 833, in stop
    ignore_live_threads=ignore_live_threads)
  File "/home/public/anaconda3/lib/python3.6/site-packages/tensorflow/python/training/coordinator.py", line 389, in join
    six.reraise(*self._exc_info_to_raise)
  File "/home/public/anaconda3/lib/python3.6/site-packages/six.py", line 693, in reraise
    raise value
  File "/home/public/anaconda3/lib/python3.6/site-packages/tensorflow/python/training/queue_runner_impl.py", line 252, in _run
    enqueue_callable()
  File "/home/public/anaconda3/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1244, in _single_operation_run
    self._call_tf_sessionrun(None, {}, [], target_list, None)
  File "/home/public/anaconda3/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1409, in _call_tf_sessionrun
    run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape mismatch in tuple component 1. Expected [513,513,3], got [513,528,3]
     [[Node: batch/padding_fifo_queue_enqueue = QueueEnqueueV2[Tcomponents=[DT_INT64, DT_FLOAT, DT_STRING, DT_INT32, DT_UINT8, DT_INT64], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/device:CPU:0"](batch/padding_fifo_queue, Reshape_3/_1645, add_2/_1647, ParseSingleExample/ParseSingleExample:1, add_3/_1649, batch/packed, Reshape_6/_1651)]]

Как я могу решить эту проблему, возникшую при прогнозировании вывода с использованием модели Deeplab?Спасибо!

...