読者です 読者をやめる 読者になる 読者になる

SuprSonicJetBoy's blog

いろいろです。

TensorFlow

TensorFlowでランダムな数値を作成する

TensorFlowでランダムな数値を作るのに手こずったのでメモ。 方法 _n = tf.random_uniform([1], minval=0, maxval=4, dtype=tf.int32, seed=None, name=None) n = _n[0] import numpy as np _n = tf.random_shuffle(np.arange(4), seed=None, name=None) n =…

TensorFlow の CIFAR-10で実際に予測してみる

TensorFlowのCIFAR-10のチュートリアルを最後まで終えると、学習済みデータをテストデータで評価することができます。 その次の段階としては、実際に学習済みデータを使って、入力された画像の予測ラベルを出力できると実用的なものとなります。 入力画像に…

TensorFlowをGPUで回すと落ちる場合の改善策

TensorFlowをGPUで回すとPCが落ちてしまうので、仕方なくCPUで回していたのですが、どうにもこうにも遅すぎるし、せっかく対応しているGPUがあるので、ちゃんと回るように対策をしてみました。 症状 TensorFlowのチュートリアル「CIFAR-10」をGPUで回すと、…

TensorFlowで画像をTFRecordsで読み書きしてみる

TensorFlowでCNNを試しているのですが、教師データのJPEG画像をTFRecords形式への変換と読み書きに四苦八苦したので、そのメモ。 学習時にいろいろな画像サイズで試したくて、1枚の画像につき、 32x32 64x64 96x96 128x128 をそれぞれ用意していたのですが、…