手势demo在do_quantization=True量化出错

--> Loading model
done
--> Building model
E Catch exception when building RKNN model!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 498, in rknn.api.rknn_base.RKNNBase.build
T   File "rknn/api/rknn_base.py", line 1071, in rknn.api.rknn_base.RKNNBase._quantize
T   File "rknn/base/RKNNlib/app/tensorzone/quantization.py", line 144, in rknn.base.RKNNlib.app.tensorzone.quantization.Quantization.run
T   File "rknn/base/RKNNlib/app/tensorzone/quantization.py", line 91, in rknn.base.RKNNlib.app.tensorzone.quantization.Quantization._run_quantization
T   File "rknn/base/RKNNlib/app/tensorzone/workspace.py", line 167, in rknn.base.RKNNlib.app.tensorzone.workspace.Workspace._setup_graph
T   File "rknn/base/RKNNlib/app/tensorzone/workspace.py", line 168, in rknn.base.RKNNlib.app.tensorzone.workspace.Workspace._setup_graph
T   File "rknn/base/RKNNlib/app/tensorzone/graph.py", line 59, in rknn.base.RKNNlib.app.tensorzone.graph.Graph.generate
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 269, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 273, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 300, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 300, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 300, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 326, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
T   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 331, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
T   File "rknn/base/RKNNlib/layer/RKNNlayer.py", line 280, in rknn.base.RKNNlib.layer.RKNNlayer.RKNNLayer.compute_tensor
T   File "rknn/base/RKNNlib/layer/dropout.py", line 20, in rknn.base.RKNNlib.layer.dropout.Dropout.compute_out_tensor
T   File "/home/mifpay/.local/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
T     return func(*args, **kwargs)
T   File "/home/mifpay/.local/lib/python3.6/site-packages/tensorflow/python/ops/nn_ops.py", line 2979, in dropout
T     return dropout_v2(x, rate, noise_shape=noise_shape, seed=seed, name=name)
T   File "/home/mifpay/.local/lib/python3.6/site-packages/tensorflow/python/ops/nn_ops.py", line 3024, in dropout_v2
T     "range [0, 1), got %g" % rate)
T ValueError: rate must be a scalar tensor or a float in the range [0, 1), got 1
done
E RKNN model data is None, please load model first!

do_quantization=False时则没有问题

whxhngns200 发表于 2019-5-21 10:39
do_quantization=False时则没有问题

看log是tensorflow的报错，估计是量化时tensorflow运行异常，可能跟网络结构有关