在iOS上實現TensorFlow Attention OCR

Question

我已經成功地訓練（使用Inception V3權重作為初始化）此處描述的Attention OCR模型： https : //github.com/tensorflow/models/tree/master/attention_ocr並將生成的檢查點文件凍結為圖形。 如何在iOS上使用C ++ API來實現此網絡？

先感謝您。

Answer 1

正如其他人建議你可以使用一些現有的iOS演示（ 1 ， 2 ）為起點，但要密切注意以下細節：

1. 確保使用正確的工具“凍結”模型。 SavedModel是Tensorflow模型的通用序列化格式。
2. 模型導出腳本通常可以執行某種輸入歸一化。 請注意，Model.create_base函數期望形狀為[batch_size，高度，寬度，通道]的tf.float32張量，其值標准化為[-1.25，1.25] 。 如果您將圖像歸一化作為TensorFlow計算圖的一部分，請確保圖像未經歸一化傳遞，反之亦然。

3. 要獲取輸入/輸出張量的名稱，您可以簡單地打印它們，例如在導出腳本中的某個位置：

    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)