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

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

KerasデータセットのCIFAR-100画像とクラス名を表示する簡単サンプル

f:id:uchidamax:20180116114500p:plain CIFAR-10画像の表示を作ったついでに、CIFAR-100画像の表示も作っておこうかと作りました。

CIFAR-100とは

一般物体認識のベンチマークとしてよく使われている画像データセット

特徴

  • 画像サイズは32ピクセルx32ピクセル
  • 全部で60000枚
  • 50000枚(各クラス5000枚)の訓練画像と10000枚(各クラス1000枚)のテスト画像に分割されている
  • クラスラベルは100クラスあり、それぞれ600枚づつ画像がある。
  • クラスラベルはスーパークラス20個とクラス100個。 詳細は下記。
スーパークラス クラス
海洋哺乳類 ビーバー, イルカ, カワウソ, アザラシ, 鯨
魚類 観賞魚, カレイ, エイ, サメ, マス
ラン, ひなげし, バラ, ヒマワリ, チューリップ
食品容器 ボトル, ボウル, カン, カップ, プレート
果物と野菜 リンゴ, キノコ, オレンジ, ナシ, ピーマン
家庭用電気機器 時計, コンピュータ・キーボード, ランプ, 電話, テレビ
家庭用家具 ベッド, 椅子, ソファー, テーブル, タンス
昆虫 蜂, カブトムシ, 蝶, 蝶等の幼虫, ゴキブリ
大きな肉食獣 クマ, ヒョウ, ライオン, 虎, 狼
大きな人造の屋外物 橋, 城, 家, 道, 超高層ビル
大きな自然野外シーン 雲, 森, 山, 平野, 海
大きな雑食動物と草食動物 ラクダ, 牛, チンパンジー, 象, カンガルー
中位の哺乳類 キツネ, ヤマアラシ, オポッサム, アライグマ, スカンク
昆虫でない無脊椎動物 カニ, ロブスター, カタツムリ, クモ, ワーム
人々 赤ちゃん, 少年, 少女, 男性, 女性
爬虫類 ワニ, 恐竜, トカゲ, ヘビ, 亀
小哺乳類 ハムスター, マウス, ウサギ, トガリネズミ, リス
カエデ, オーク, ヤシ, 松, 柳
乗り物 1 自転車, バス, オートバイ, ピックアップトラック, 電車
乗り物 2 芝刈り機, ロケット, 市街電車, タンク, トラク

CIFAR-100の画像とクラス名を表示するソースコード

jupyter notebook 

でjupyter notebookを起動してから使ってください。

参考

次のサイトを参考にさせていただきました。
感謝!!

tensorflow.classcat.com
CIFAR-100のクラス分類が一通り日本語でのってる!

corochann.com CIFAR-100の100個のクラス名の配列がのってて素晴らしい!

データセット - Keras Documentation