我尝试使用pb
将tblite
转换为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上看到它,请告诉我。