Могу ли я перевести все тонкие модели tenorflow в tflite?

Question

Я обучаю тонкие модели основе tenorflow для классификации изображений в пользовательских наборах данных.Прежде чем уделять много времени обучению такому огромному набору данных, я хотел узнать, могу ли я преобразовать все модели, доступные в slim model zoo , в формат tflite.

Кроме того,Я знаю, что я могу преобразовать свою собственную Slim-модель в замороженный график.Это шаг после этого, о котором я беспокоюсь, т. Е. Преобразование в .tflite из моей специально обученной модели .pb.

Поддерживается ли это?или есть кто-то, кто сталкивается с проблемами конверсии, которые еще не решены?

Спасибо.

Answer 1

Многие модели Slim можно преобразовать в TFLite, но это не гарантия, поскольку некоторые модели могут иметь операции, не поддерживаемые TFLite.

Что вы можете сделать, так это попробовать преобразовать вашу модель в TensorFlow Liteиспользование TFLiteConverter в Python перед тренировкой.Если преобразование выполнено успешно, вы можете обучить свою модель TF и ​​преобразовать ее еще раз.