我试着把我的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在控制台中打印
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但是我的临时文件夹是空的。
发布于 2019-01-22 13:45:34
问题可能出在这里
--output_file=tmp/opt_mask_rcnn_New.tflite \试一试
--output_file=/tmp/opt_mask_rcnn_New.tflite \或者另一条绝对路径
https://stackoverflow.com/questions/54285708
复制相似问题