tenorflow Невозможно преобразовать тензор ресурса dtype в массив NumPy. Tflite - PullRequest
Новая
       0

tenorflow Невозможно преобразовать тензор ресурса dtype в массив NumPy. Tflite

1 голос
/ 27 мая 2020

Я обучил сеть (git ссылка ниже) и сохранил в формате сохраненной модели. И хотел бы преобразовать его в tflite. Я использую python API для tflite coverter (tflite.py в git ссылка ниже). Но я не могу это сделать.

Информация о системе:
Платформа и распространение ОС: Ubuntu 18.04.3 LTS
Версия TensorFlow: tenorflow / tensorflow: 2.2.0-gpu (docker)

Ссылка на сеть и код модели сохранения.

Результат вызова конвертера: 

File "tflite.py", line 22, in convert_model
  tflite_model = converter.convert()
File "/usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/lite.py", line 459, in convert
  self._funcs[0], lower_control_flow=False))
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/convert_to_constants.py", line 706, in convert_variables_to_constants_v2_as_graph
  func, lower_control_flow, aggressive_inlining)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/convert_to_constants.py", line 457, in _convert_variables_to_constants_v2_impl
  tensor_data = _get_tensor_data(func)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/convert_to_constants.py", line 217, in _get_tensor_data
  data = val_tensor.numpy()
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 961, in numpy
  maybe_arr = self._numpy()  # pylint: disable=protected-access
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 929, in _numpy
  six.raise_from(core._status_to_exception(e.code, e.message), None)
File "<string>", line 3, in raise_from
tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot convert a Tensor of dtype resource to a NumPy array.

Ссылка на сохраненную модель

...