---------------------------------------------------------------------------
UnknownError Traceback (most recent call last)
<ipython-input-10-f100b465891f> in <module>
14 # 이미지에 대해 Object Detection 수행.
15 start = time.time()
---> 16 boxes, scores, labels = retina_model.predict_on_batch(np.expand_dims(image, axis=0))
17 print(boxes.shape, scores.shape, labels.shape)
18 print("processing time: ", time.time() - start)
~/.conda/envs/tf115/lib/python3.6/site-packages/keras/engine/training.py in predict_on_batch(self, x)
1578 ins = x
1579 self._make_predict_function()
-> 1580 outputs = self.predict_function(ins)
1581 return unpack_singleton(outputs)
1582
~/.conda/envs/tf115/lib/python3.6/site-packages/tensorflow_core/python/keras/backend.py in __call__(self, inputs)
3474
3475 fetched = self._callable_fn(*array_vals,
-> 3476 run_metadata=self.run_metadata)
3477 self._call_fetch_callbacks(fetched[-len(self._fetches):])
3478 output_structure = nest.pack_sequence_as(
~/.conda/envs/tf115/lib/python3.6/site-packages/tensorflow_core/python/client/session.py in __call__(self, *args, **kwargs)
1470 ret = tf_session.TF_SessionRunCallable(self._session._session,
1471 self._handle, args,
-> 1472 run_metadata_ptr)
1473 if run_metadata:
1474 proto_data = tf_session.TF_GetBuffer(run_metadata_ptr)
UnknownError: 2 root error(s) found.
(0) Unknown: 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 conv1/convolution}}]]
[[filtered_detections/map/while/Switch_1/_1877]]
(1) Unknown: 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 conv1/convolution}}]]
0 successful operations.
0 derived errors ignored.
