Преобразование модели keras в модель tflite в ноутбуке Jupyter - PullRequest
Новая
       4

Преобразование модели keras в модель tflite в ноутбуке Jupyter

0 голосов
/ 13 февраля 2020

Я сохранил модель keras со следующей строкой кода 

keras.save("saved_model.h5")

Когда я пытаюсь оптимизировать и конвертировать в модель .tflite, используя этот код 

converter=tt.lite.TFLiteConverter.from_keras_model_file("saved_model.h5")
converter.optimizations = [tf.lite.Optimize.DEFAULT]
tflite_model = converter.convert()

Я получаю это ошибка 

ConverterError                            Traceback (most recent call last)
<ipython-input-23-faeb309a4cbc> in <module>()
----> 1 tflite_model = converter.convert()
      2 with open("converted_model.tflite", "wb") as tffile:
      3     tffile.write(tflite_model)

~/.local/lib/python3.6/site-packages/tensorflow/lite/python/lite.py in

convert (self) 896 input_tensors = self._input_tensors, 897 output_tensors = self._output_tensors, -> 898 ** converter_kwargs) 899 else: 900 result = _toco_convert_graph_def ( 

~/.local/lib/python3.6/site-packages/tensorflow/lite/python/convert.py

в toco_convert_impl (input_data, input_tensors, output_tensors, * args, ** kwargs) 402 data = toco_convert_protos (model_flags.SerializeToString (), 403 toco_flags.SerializeToString (), -> 404 input ()) 405 возвращают данные 406 

~/.local/lib/python3.6/site-packages/tensorflow/lite/python/convert.py

в toco_convert_protos (model_flags_str, toco_flags_str, input_data_str) 170 stderr = _try_convert_to_unicode (stderr) 171 вызывают ConverterError (-> 172 "информация о TOCO не выполнена, см. Консоль. n% s \ n% s \ n "% (stdout, stderr)) 173 наконец: 174 # Необходимо вручную очистить файлы. 

ConverterError: TOCO failed. See console for info.
/home/scripter/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:516:

FutureWarning: Передача (тип, 1) или '1тип' a s синоним типа устарел; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_qint8 = np.dtype ([("qint8", np.int8, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py: 517: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_quint8 = np.dtype ([("quint8", np.uint8, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py: 518: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_qint16 = np.dtype ([("qint16", np.int16, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py: 519: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_quint16 = np.dtype ([("quint16", np.uint16, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py: 520: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (тип, (1,)) / '(1,) тип'. _np_qint32 = np.dtype ([("qint32", np.int32, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py: 525: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (тип, (1,)) / '(1,) тип'. np_resource = np.dtype ([("resource", np.ubyte, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py: 541: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_qint8 = np.dtype ([("qint8", np.int8, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py: 542: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_quint8 = np.dtype ([("quint8", np.uint8, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py: 543: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа осуждается; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_qint16 = np.dtype ([("qint16", np.int16, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py: 544: FutureWarning: передача (type, 1) или '1type' в качестве синонима типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_quint16 = np.dtype ([("quint16", np.uint16, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py: 545: FutureWarning: передача (type, 1) или '1type' как синоним типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. _np_qint32 = np.dtype ([("qint32", np.int32, 1)]) /home/scripter/.local/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py: 550: FutureWarning: передача (type, 1) или '1type' как синоним типа устарела; в будущей версии numpy это будет пониматься как (type, (1,)) / '(1,) type'. np_resource = np.dtype ([("resource", np.ubyte, 1)]) Traceback (последний вызов был последним): файл "/home/scripter/.anaconda3/bin/toco_from_protos", строка 7, из тензорного потока. contrib.lite.toco. python .toco_from_protos import main ModuleNotFoundError: Нет модуля с именем 'tenorflow.contrib.lite'

Я использую tenorflow версии 1.14.0. Что я могу делать не так

...