Внедрение TensorFlow Attention OCR на iOS
Я успешно обучил (используя веса Inception V3 в качестве инициализации) модель распознавания внимания, описанную здесь: https://github.com/tensorflow/models/tree/master/attention_ocr и заморозил полученные файлы контрольных точек в граф. Как реализовать эту сеть с помощью API C ++ на iOS?
Заранее спасибо.
Решение
Как предлагают другие, вы можете использовать некоторые существующие демоверсии iOS (1, 2) в качестве отправной точки, но обратите пристальное внимание на следующие детали:
- Убедитесь, что вы используете правильные инструменты, чтобы «заморозить» модель. SavedModel универсальный формат сериализации для моделей Tensorflow.
- Сценарий экспорта модели может и обычно выполняет некоторую нормализацию ввода. Обратите внимание, что функция Model.create_base ожидает тензор tf.float32 формы [batch_size, height, width, channel] со значениями, нормализованными до [-1.25, 1.25]. Если вы выполняете нормализацию изображения как часть графика вычислений TensorFlow, убедитесь, что изображения переданы ненормализовано, и наоборот.
-
Чтобы получить имена тензоров ввода / вывода, вы можете просто напечатать их, например, где-то в вашем скрипте экспорта:
data_images = tf.placeholder(dtype=tf.float32, shape=[batch_size, height, width, channels], name='normalized_input_images') endpoints = model.create_base(data_images, labels_one_hot=None) print(data_images, endpoints.predicted_chars, endpoints.predicted_scores)
Другие решения
Других решений пока нет …