|
发表于 2020-11-24 14:49:48
|
显示全部楼层
本帖最后由 李雪梅 于 2020-11-24 14:50 编辑
依据转换为RKNN模型代码我的会出现如下报错,请楼主赐教:
xml@xml-vm:~/project/rk3399ProD/RK3399Pro入门教程(4)$ python3 tf_pb-to-rknn.py
--> config model
done
--> Loading model
/home/xml/.local/lib/python3.6/site-packages/onnx_tf/common/__init__.py:87: UserWarning: FrontendHandler.get_outputs_names is deprecated. It will be removed in future release.. Use node.outputs instead.
warnings.warn(message)
done
--> Building model
W The channel_mean_value filed will not be used in the future!
W The channel_mean_value of input 0 is not support, ignore it!
E Catch exception when building RKNN model!
Exception in thread Thread-1:
Traceback (most recent call last):
File "/usr/lib/python3.6/threading.py", line 916, in _bootstrap_inner
self.run()
File "rknn/base/acuitylib/provider/queue_provider.py", line 98, in rknn.base.acuitylib.provider.queue_provider.QueueProvider.run
File "rknn/base/acuitylib/provider/queue_provider.py", line 102, in rknn.base.acuitylib.provider.queue_provider.QueueProvider.run
File "rknn/base/acuitylib/provider/text_provider.py", line 46, in rknn.base.acuitylib.provider.text_provider.TextProvider.get_batch
File "rknn/base/acuitylib/provider/file_path_provider.py", line 98, in rknn.base.acuitylib.provider.file_path_provider.FilePathProvider._data_to_tensor
File "rknn/base/acuitylib/provider/file_path_provider.py", line 110, in rknn.base.acuitylib.provider.file_path_provider.FilePathProvider._fitting_scale
IndexError: list index out of range
E Traceback (most recent call last):
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1278, in _do_call
E return fn(*args)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1263, in _run_fn
E options, feed_dict, fetch_list, target_list, run_metadata)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1350, in _call_tf_sessionrun
E run_metadata)
E tensorflow.python.framework.errors_impl.OutOfRangeError: FIFOQueue '_0_fifo_queue' is closed and has insufficient elements (requested 1, current size 0)
E [[Node: fifo_queue_Dequeue = QueueDequeueV2[component_types=[DT_FLOAT], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/device:CPU:0"](fifo_queue)]]
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 895, in rknn.api.rknn_base.RKNNBase.build
E File "rknn/api/rknn_base.py", line 1797, in rknn.api.rknn_base.RKNNBase._quantize2
E File "rknn/base/RKNNlib/app/medusa/quantization.py", line 105, in rknn.base.RKNNlib.app.medusa.quantization.Quantization.run
E File "rknn/base/RKNNlib/app/medusa/quantization.py", line 44, in rknn.base.RKNNlib.app.medusa.quantization.Quantization._run_quantization
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 145, in rknn.base.RKNNlib.app.medusa.workspace.Workspace.run
E File "rknn/base/RKNNlib/app/medusa/workspace.py", line 126, in rknn.base.RKNNlib.app.medusa.workspace.Workspace._run_iteration
E File "rknn/base/RKNNlib/RKNN_session.py", line 30, in rknn.base.RKNNlib.RKNN_session.RKNNSession.run
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 877, in run
E run_metadata_ptr)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1100, in _run
E feed_dict_tensor, options, run_metadata)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1272, in _do_run
E run_metadata)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1291, in _do_call
E raise type(e)(node_def, op, message)
E tensorflow.python.framework.errors_impl.OutOfRangeError: FIFOQueue '_0_fifo_queue' is closed and has insufficient elements (requested 1, current size 0)
E [[Node: fifo_queue_Dequeue = QueueDequeueV2[component_types=[DT_FLOAT], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/device:CPU:0"](fifo_queue)]]
E Caused by op 'fifo_queue_Dequeue', defined at:
E File "tf_pb-to-rknn.py", line 28, in <module>
E ret = rknn.build(do_quantization=True, dataset='./dataset.txt') # 量化模型
E File "/home/xml/.local/lib/python3.6/site-packages/rknn/api/rknn.py", line 249, in build
E ret = self.rknn_base.build(do_quantization=do_quantization, dataset=dataset, pack_vdata=pre_compile, batch_size=rknn_batch_size)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/ops/data_flow_ops.py", line 433, in dequeue
E self._queue_ref, self._dtypes, name=name)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/ops/gen_data_flow_ops.py", line 3734, in queue_dequeue_v2
E timeout_ms=timeout_ms, name=name)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
E op_def=op_def)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 454, in new_func
E return func(*args, **kwargs)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3155, in create_op
E op_def=op_def)
E File "/home/xml/.local/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1717, in __init__
E self._traceback = tf_stack.extract_stack()
E OutOfRangeError (see above for traceback): FIFOQueue '_0_fifo_queue' is closed and has insufficient elements (requested 1, current size 0)
E [[Node: fifo_queue_Dequeue = QueueDequeueV2[component_types=[DT_FLOAT], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/device:CPU:0"](fifo_queue)]]
Build model failed!
xml@xml-vm:~/project/rk3399ProD/RK3399Pro入门教程(4)$
|
|