0

我正在解决一个问题，我使用自定义数据集使用 Tensorflow 对象检测 API 训练模型。我正在使用 tf 版本 2.2.0

output_directory = 'inference_graph'
!python /content/models/research/object_detection/exporter_main_v2.py \
--trained_checkpoint_dir {model_dir} \
--output_directory {output_directory} \
--pipeline_config_path {pipeline_config_path}

我能够成功获得 .pb 文件以及 .ckpt 文件。但现在我需要将其转换为 .tflite。我无法这样做，有一些错误或其他。

我尝试了写在 TensorFlow 文档上的基本方法，但也没有用。我尝试的另一个代码如下：

    import tensorflow as tf
from tensorflow.keras.models import Sequential, Model
from tensorflow.keras.layers import Conv2D, Flatten, MaxPooling2D, Dense, Input, Reshape, Concatenate, GlobalAveragePooling2D, BatchNormalization, Dropout, Activation, GlobalMaxPooling2D
from tensorflow.keras.utils import Sequence

model = tf.saved_model.load(f'/content/drive/MyDrive/FINAL DNET MODEL/inference_graph/saved_model/')
converter = tf.lite.TFLiteConverter.from_keras_model(model)
converter.post_training_quantize=True
converter.inference_type=tf.uint8
tflite_model = converter.convert()
open("val_converted_model_int8.tflite", "wb").write(tflite_model)

我得到的错误是：

() 中的 AttributeError Traceback (最近一次调用最后一次) 8 converter.post_training_quantize=True 9 converter.inference_type=tf.uint8 ---> 10 tflite_model = converter.convert() 11 open("val_converted_model_int8.tflite", "wb") .write(tflite_model)

/usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/lite.py in convert(self) 837 # to None. 838 # 一旦我们对动态形状有了更好的支持，我们就可以删除它。--> 839 if not isinstance(self._keras_model.call, _def_function.Function): 840 # Passkeep_original_batch_size=True将确保我们得到一个输入 841 # 签名，包括用户指定的批处理维度。

AttributeError：“_UserObject”对象没有属性“调用”

谁能帮我解决这个问题？

4

1 回答 1

1

我认为问题不在于可变输入形状（而错误消息令人困惑）。

tf.saved_model.load返回 a SavedModel，但tf.lite.TFLiteConverter.from_keras_model需要一个 Keras 模型，因此它无法处理它。

您需要使用TFLiteConverter.from_saved_model API。像这样的东西：

saved_model_dir = '/content/drive/MyDrive/FINAL DNET MODEL/inference_graph/saved_model/'
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)

如果您遇到其他问题，请告诉我们。

于 2021-02-12T18:52:40.743 回答