获取检查失败:当pb到tflite时,input_shapes.size()== model_flags-> input_arrays_size()

时间:2019-01-18 08:11:05

标签: tensorflow tensorflow-lite

我尝试使用pbtblite转换为bazel run tensorflow/lite/toco:toco

bazel run tensorflow/lite/toco:toco -- \
--input_file=opt_mask_rcnn_New.pb \
--output_file=$opt_mask_rcnn_New.tflite \
--input_shapes=1,2048,2048,3 \
--input_arrays=input_image,input_image_meta,input_anchors\
--output_arrays=output_detections,output_mrcnn_class,output_mrcnn_bbox,output_mrcnn_mask,output_rois,output_rpn_class,output_rpn_bbox \
--inference_type=FLOAT \

我提出了这个错误F tensorflow/lite/toco/model_cmdline_flags.cc:321] Check failed: input_shapes.size() == model_flags->input_arrays_size() Abort trap: 6

我认为参数中的input_shapes是错误的。

但是我不知道在哪里可以找到input_shapes。 如果可以在Tensorboard上看到它,请告诉我。

0 个答案:

没有答案