人工知能と競プロやってくブログ

深層学習・機械学習・AI・atcoder・競技プログラミングについて調べてやってみたことをまとめるブログです

CIFAR-10画像のラベル予測を一つ一つ表示する簡単サンプル

https://gyazo.com/b209037ac23ee815069ca28214db0669

画像ラベル予測をまとめて実行した結果の成績は良くサンプルコードで目にしますが、一つ一つの結果とどれを間違ったのかがよくわからないので作ってみました。
認識を間違った部分が赤文字で表示され、左にハズレの予測ラベル、右に正解のラベルが表示されます。

主要部分のソースコード

jupyter notebookを使っていますので

jupyter notebook 

を実行してから使ってください。

GitHubリポジトリ

github.com 予測に使用しているモデルファイルは、こちらに含まれています。