日々精進

新しく学んだことを書き留めていきます

Tensor flowのTensorオブジェクトをndarrayに変換する

以下のようにSessionを開始してTensor.eval()を実行すればいい。

sess = tf.Session()
with sess.as_default():
    X_train_color = X_train_tensor.eval()
    X_test_color = X_test_tensor.eval()
    X_valid_color = X_valid_tensor.eval()

Tensor flowのresizeなどのメソッドを使うと結果がTensorになってこういう変換が必要になるのは面倒だなぁ。。 前処理の最後とかでそのままfitに渡せば良いとかならまだいいんだろうけど。

参考: olgabelitskaya.github.io