TensorFlow - CIFAR-10の畳み込みニューラルネットワークのチュートリアルをやってみる
チュートリアルのページは、これです。日本語化してくれてるページはこれです。クロスキャットすげー。
やることとしては、MNISTの畳み込みとほぼ同じな気がする。カラー画像になって、画像が複雑化してるけど、畳み込みニューラルネットワークを使って10個に分類する点は同じ。だから基本は同じだけど、色々な工夫が入ってるのかなと思います。ソースコードはここにあります。
各ファイルは下記のようになってます。自分のPCはGPU1個なので、cifar10_train.pyを使います。
| cifar10_input.py | native の CIFAR-10 バイナリ・ファイル・フォーマットを読む。 |
|---|---|
| cifar10.py | CIFAR-10 モデルの構築。 |
| cifar10_train.py | CPU または GPU で CIFAR-10 モデルを訓練する。 |
| cifar10_multi_gpu_train.py | 複数の GPU で CIFAR-10 モデルを訓練する。 |
| cifar10_eval.py | CIFAR-10 モデルの予測性能を評価する。 |
CIFAR-10 バイナリ・ファイル・フォーマットを読む
単純に読み込むだけじゃないようです。トリミングしたり、裏返したり、白くしたり、コントラスト変えたりするそうです。