tensorflow - 使用toco将pb转换为tflite时，toco成功了，但是没有结果

Question

我试图将我的 pb 文件转换为 tflite。toco 已显示成功 我没有结果。

使用代码

bazel run tensorflow/lite/toco:toco -- \
--input_file=opt_mask_rcnn_New.pb \
--output_file=tmp/opt_mask_rcnn_New.tflite \
--input_shape=1,${IMAGE_SIZE},${IMAGE_SIZE},3:1,93:1,1,4 \
--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 \
--output_format=TFLITE \
--inference_input_type=FLOAT

在控制台中打印 toco

INFO: Analysed target //tensorflow/lite/toco:toco (0 packages loaded, 0 targets configured).
INFO: Found 1 target...
Target //tensorflow/lite/toco:toco up-to-date:
bazel-bin/tensorflow/lite/toco/toco
INFO: Elapsed time: 0.294s, Critical Path: 0.00s
INFO: 0 processes.
INFO: Build completed successfully, 1 total action
INFO: Running command line: bazel-bin/tensorflow/lite/toco/toco '--input_file=opt_mask_rcnn_New.pb' '--output_file=tmp/opt_mask_rcnn_New.tflite' '--input_shape=1,,,3:1,93:1,1,4' '--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' 
'--oINFO: Build completed successfully, 1 total action

但是我的 tmp 文件夹是空的。

标签： tensorflowtensorflow-lite

Answer 1

问题大概就在这里

--output_file=tmp/opt_mask_rcnn_New.tflite \

尝试

--output_file=/tmp/opt_mask_rcnn_New.tflite \

或者其他一些绝对路径