search

senguptaumd / Background-Matting

Background Matting: The World is Your Green Screen
https://grail.cs.washington.edu/projects/background-matting/
4.78k stars 662 forks source link

a error when the segmentation (Deeplab) and matting run together. #50

Open sundyll opened 4 years ago

sundyll commented 4 years ago

Hi I load the models of segmentation and matting at the same time, and hope to get the segmentation img by the segmentation net, and matting by the matting net. But:

2020-05-25 11:06:13.415797: E tensorflow/stream_executor/cuda/cuda_dnn.cc:324] Loaded runtime CuDNN library: 7.1.2 but source was compiled with: 7.4.2. CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library. If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration. 2020-05-25 11:06:13.418751: E tensorflow/stream_executor/cuda/cuda_dnn.cc:324] Loaded runtime CuDNN library: 7.1.2 but source was compiled with: 7.4.2. CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library. If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration. Traceback (most recent call last): File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1334, in _do_call return fn(*args) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1319, in _run_fn options, feed_dict, fetch_list, target_list, run_metadata) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1407, in _call_tf_sessionrun run_metadata) tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[{{node MobilenetV2/Conv/Conv2D}}]] [[{{node SemanticPredictions}}]]

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "demo.py", line 10, in img_seg = P.pred(img) File "/home/sundy/Background-Matting/pred_seg.py", line 162, in pred res_im,seg = self.MODEL.run(image) File "/home/sundy/Background-Matting/pred_seg.py", line 70, in run feed_dict={self.INPUT_TENSOR_NAME: [np.asarray(resized_image)]}) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 929, in run run_metadata_ptr) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1152, in _run feed_dict_tensor, options, run_metadata) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1328, in _do_run run_metadata) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1348, in _do_call raise type(e)(node_def, op, message) tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[node MobilenetV2/Conv/Conv2D (defined at /home/sundy/Background-Matting/pred_seg.py:49) ]] [[node SemanticPredictions (defined at /home/sundy/Background-Matting/pred_seg.py:49) ]]

Caused by op 'MobilenetV2/Conv/Conv2D', defined at: File "demo.py", line 6, in P = pred_seg() File "/home/sundy/Background-Matting/pred_seg.py", line 154, in init self.MODEL = DeepLabModel(download_path) File "/home/sundy/Background-Matting/pred_seg.py", line 49, in init tf.import_graph_def(graph_def, name='') File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func return func(*args, **kwargs) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/framework/importer.py", line 442, in import_graph_def _ProcessNewOps(graph) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/framework/importer.py", line 235, in _ProcessNewOps for new_op in graph._add_new_tf_operations(compute_devices=False): # pylint: disable=protected-access File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3433, in _add_new_tf_operations for c_op in c_api_util.new_tf_operations(self) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3433, in for c_op in c_api_util.new_tf_operations(self) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3325, in _create_op_from_tf_operation ret = Operation(c_op, self) File "/home/sundy/.local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 1801, in init self._traceback = tf_stack.extract_stack()

UnknownError (see above for traceback): Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[node MobilenetV2/Conv/Conv2D (defined at /home/sundy/Background-Matting/pred_seg.py:49) ]] [[node SemanticPredictions (defined at /home/sundy/Background-Matting/pred_seg.py:49) ]]