推論のためにグラフを RT にエクスポートしたい。保存されたモデルでそれを達成する方法。保存されたモデルを使用した推論のコードは次のとおりです。
import tensorflow as tf
import cv2
img = cv2.imread("test.jpg")
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img_rgb = np.expand_dims(img, 0)
predict_fn = tf.contrib.predictor.from_saved_model("./saved_model")
output_data = predict_fn({"inputs": img_rgb})
print(output_data) # detector output dictionary
使いたくないfrozen_inference_garph.pb
。推論に使用したいsaved_model/saved_model.pb
(その方が精度が高いので)。