預測張量流對象檢測中的單個圖像

Question

我已經使用tensorflow object detection存儲庫訓練了對象檢測模型。

現在在評估default way是從record文件中使用tf.contrib.slim.parallel_reader

我想讀直接使用文件系統中的圖像，並獲得圖像的預測閱讀。我怎麼做？

Answer 1

看一看這個https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb，這裏使用了一個冷凍模型，但是你可以使用普通的ckpt，只需改變一點邏輯就可以使用圖像而不是tfrecord。

第一步：使用CKPT文件加載模型

第二步：獲取模型的端點運行的預測;類似上面的鏈接，除了去除:0

第三步：使用佔位符喂numpy的數組（您的圖像）

從這個

除了可以訓練的模型轉換成冷凍模式convert to frozen model，並按照相同的步驟在上面的鏈接中描述。