Проблема при попытке экспортировать мою модель Tensor Flow в TLITE
После обучения моей модели в GoogleColab, следуя этому руководству, когда я пытаюсь запустить model.export('image_classifier.tflite', 'image_labels.txt')
это показывает мне
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-37-4d5419f8b12d> in <module>()
----> 1 model.export('image_classifier.tflite', 'image_labels.txt')
1 frames
/usr/local/lib/python3.6/dist-packages/tensorflow_examples/lite/model_customization/core/task/image_classifier.py in export(self, tflite_filename, label_filename, **kwargs)
185 else:
186 quantized = False
--> 187 self._export_tflite(tflite_filename, label_filename, quantized)
188 else:
189 raise ValueError('Model Export Format %s is not supported currently.' %
/usr/local/lib/python3.6/dist-packages/tensorflow_examples/lite/model_customization/core/task/classification_model.py in _export_tflite(self, tflite_filename, label_filename, quantized)
130 quantized: boolean, if True, save quantized model.
131 """
--> 132 converter = tf.lite.TFLiteConverter.from_keras_model(self.model)
133 if quantized:
134 converter.optimizations = [tf.lite.Optimize.OPTIMIZE_FOR_SIZE]
AttributeError: type object 'TFLiteConverter' has no attribute 'from_keras_model
1 ответ
Решение
Проблема решена обновлением версии Tensorflow, я использовал версию 1.X, и TensorFlow сильно изменился в новой версии ЕСЛИ ВЫ ИСПОЛЬЗУЕТЕ COLAB ИЗ GOOGLE, версия все еще 1.X и скоро изменится на 2, поэтому попробуйте Постройте свою модель другим способом.