转换 时出错.Pb filte to tflite using toco(tensorflow).



我正在为诗人遵循本教程的张量流,我创建自定义训练数据并将其转换为 .pb 但是当我尝试使用 toco 将 .pb 转换为 .tflite 时,我收到错误

2018-10-10 13:56:40.442130:I tensorflow/contrib/lite/toco/import_tensorflow.cc:1233] 转换不支持的操作:解码Jpeg 2018-10-10 13:56:40.471545:I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] 删除未使用的操作之前:706 个运算符,1102 个数组(0 个量化( 2018-10-10 13:56:40.485896:I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] 删除未使用的操作后,传递 1:607 个运算符,1003 个数组(0 个量化( 2018-10-10 13:56:40.502109:I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] 在一般图形转换之前:607 个运算符,1003 个数组(0 个量化( 2018-10-10 13:56:40.581854:I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] 一般图变换通过 1 后:129 个运算符,327 个数组(0 个量化( 2018-10-10 13:56:40.584067:I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] 一般图变换通过 2 后:126 个运算符,321 个数组(0 个量化( 2018-10-10 13:56:40.586222:I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] 一般图变换通过 3:125 个运算符,319 个数组(0 个量化( 2018-10-10 13:56:40.588344:I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] 在去量化图转换之前:125 个运算符,319 个数组(0 个量化( 2018-10-10 13:56:40.590472:I tensorflow/contrib/lite/toco/allocate_transient_arrays.cc:311] 总瞬态数组分配大小:0 字节,理论最优值:0 字节。 2018-10-10 13:56:40.591866:F tensorflow/contrib/lite/toco/tflite/export.cc:304] 标准 TensorFlow Lite 运行时不支持模型中的某些运算符。如果您有它们的自定义实现,则可以使用 --allow_custom_ops 禁用此错误。以下是您需要自定义实现的运算符列表:DecodeJpeg,ExpandDims。 已中止(核心已转储(

toco 
--input_format=TENSORFLOW_GRAPHDEF 
--graph_def_file=tf_files/object_graph.pb 
--output_format=TFLITE 
--output_file=tf_files/qobject.lite 
--inference_type=QUANTIZED_UINT8 
--inference_input_type=QUANTIZED_UINT8 
--input_arrays=input 
--output_arrays=final_result 
--input_shapes=1,224,224,3
--mean_values=128 
--std_dev_values=128 
--default_ranges_min=0 
--default_ranges_max=6

只需使用此代码将.pb文件转换为精简版,并量化

最新更新