Модель тензор потока в Numpy - PullRequest
0 голосов
/ 24 января 2019

Я хочу использовать предварительно обученную SqueezeNet модель, реализованную в Tensorflow, в другом пакете Python.Проблема в том, что текущая версия Tenorflow не поддерживает Python 3.7 , а также включает в себя весь пакет Tensorflow в качестве требования, что не является хорошей идеей, поскольку она занимает много места и может стать дополнительным источником проблем.в упаковке.Лучшим вариантом было бы избежать Tensorflow как требование.

Есть ли возможность преобразовать модель Tensorflow в операции в Numpy или любое другое более легкое представление.Мне не нужна поддержка для обучения модели, так как она уже прошла предварительную подготовку.Я буду использовать модель только для целей прогнозирования.Решение должно иметь возможность быть упакованным в пакет Python (устанавливается через pip).

Любые другие предложения приветствуются.

1 Ответ

0 голосов
/ 24 января 2019

Вы можете попробовать экспортировать свою модель в tflite и посмотреть, можете ли вы позволить себе зависеть от этого.В настоящее время не поддерживается способ автоматического преобразования любой модели tenorflow в numpy, но вы можете сделать это вручную, проверив вашу модель.

...