Tensorflow Lite不学习输出范围

时间:2019-10-09 08:22:38

标签: tensorflow-lite

我正在对车辆预测模型进行量化,看来.tflite文件实际上并未存储正确的输出范围,解释器.get_output_details()的结果如下:

output_zero_points:[0,0,0,0,0,0]

output_scales:[0.027450980618596077、0.027450980618596077、0.027450980618596077、0.027450980618596077、0.027450980618596077、0.027450980618596077]

我的模型中有6个输出张量,并且上面的范围不正确。 Bash脚本是

tflite_convert \
--output_file=${OUTDIR}/final2.tflite \
--graph_def_file=${OUTDIR}/frozen_eval2.pb \
--inference_type=${QTYPE} \
--input_type=${QTYPE} \
--input_arrays=images_ph,roadnet_ph,rand_ph,lstm_state1_ph,lstm_state2_ph,lstm_state3_ph,lstm_state4_ph,lstm_state5_ph \
--output_arrays=cdna_outs,lstm_state1,lstm_state2,lstm_state3,lstm_state4,lstm_state5 \
--mean_values=0,0,118,124,130,120,127,119 \
--std_dev_values=255,255,38,20,21,20,20,20 \
--default_ranges_min=0 \
--default_ranges_max=7 \
--target_ops=TFLITE_BUILTINS,SELECT_TF_OPS \
--change_concat_input_ranges=false

0 个答案:

没有答案