0

推論のためにグラフを 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(その方が精度が高いので)。

4

1 に答える 1