Как преобразовать модель keras в формат corml с помощью coremltools

Question

Я хочу преобразовать мою модель keras в coreml, используя coremltools. Когда я пытаюсь это сделать, выдает ошибку

ImportError: невозможно импортировать имя 'relu6'

Моя версия tenorflow 1.5.1 Моя версия keras 2.1.6

Полный файл colab находится здесь:

https://colab.research.google.com/drive/1kSeErLsp_xaU37haUrwBO5jiNlV2RCll

Я уже пробовал разные версии модулей, но я готов попробовать новую версию, которую я не пробовал

Answer 1

Похоже, что ваша установка все еще пытается использовать Keras 2.2.0, поскольку ошибка в coremltools происходит после того, как он проверит, что версия Keras>= 2.2.0.

Написать keras.__version__ чтобы узнать, какую версию Keras использует ваш ноутбук.

Попробуйте установить более старую версию keras_applicationsтот, который все еще имеет relu6 функция. Это было недавно изменено. Проблема с Keras в том, что вещи часто перемещаются между второстепенными версиями.