将SSD_ResNet_50_FPN_COCO模型转换为TFLite

Question

我正在尝试将tensorflow detection model zoo repository中的ssd_resnet_50_fpn_coco模型转换为.tflite。

档案包含：

• 已保存的模型（saved_model / saved_model.pb）
• 检查点（model.ckpt.data-00000-of-00001，model.ckpt.index，model.ckpt.meta）
• 冻结图原型（frozen_inference_graph.pb）

要进行转换，我使用Converter command line examples

中的命令行

tflite_convert \
--output_file=test.tflite \
--saved_model_dir=saved_model

哪个返回错误：

ValueError: None is only supported in the 1st dimension. Tensor 'image_tensor' has invalid shape '[None,None,None,3]'.

由于ImageTensor是input_arrays，因此我试图在命令行中添加input_shapes，但无济于事。

tflite_convert \
--output_file=test.tflite \
--saved_model_dir=saved_model \
--input_arrays=ImageTensor \
--input_shapes=1,640,640,3

但是我遇到相同的错误，将--input_arrays=ImageTensor替换为--input_arrays= image_tensor并没有改变，并返回了相同的错误：

ValueError: None is only supported in the 1st dimension. Tensor 'image_tensor' has invalid shape '[None,None,None,3]'.

在转换VGG16模型并添加input_shapes和input_arrays是技巧时，我遇到了相同的错误，但不是在这里

通过我使用tensorflow 1.13.1和1.14.1-dev（每晚构建）