AI

TensorFlow - CIFAR-10の畳み込みニューラルネットワークのチュートリアルをやってみる

チュートリアルのページは、これです。日本語化してくれてるページはこれです。クロスキャットすげー。

やることとしては、MNISTの畳み込みとほぼ同じな気がする。カラー画像になって、画像が複雑化してるけど、畳み込みニューラルネットワークを使って10個に分類する点は同じ。だから基本は同じだけど、色々な工夫が入ってるのかなと思います。ソースコードはここにあります。

各ファイルは下記のようになってます。自分のPCはGPU1個なので、cifar10_train.pyを使います。

cifar10_input.pynative の CIFAR-10 バイナリ・ファイル・フォーマットを読む。
cifar10.pyCIFAR-10 モデルの構築。
cifar10_train.pyCPU または GPU で CIFAR-10 モデルを訓練する。
cifar10_multi_gpu_train.py複数の GPU で CIFAR-10 モデルを訓練する。
cifar10_eval.pyCIFAR-10 モデルの予測性能を評価する。

CIFAR-10 バイナリ・ファイル・フォーマットを読む

単純に読み込むだけじゃないようです。トリミングしたり、裏返したり、白くしたり、コントラスト変えたりするそうです。