コンテンツへスキップ

ImageNet10データセット

ImageNet10データセットは、ImageNetデータベースの小規模なサブセットである。 Ultralyticsによって開発されたImageNetデータベースの小規模なサブセットであり、CIテスト、サニティチェック、トレーニングパイプラインの高速テスト用に設計されている。このデータセットは、トレーニングセットの最初の画像と、ImageNetの最初の10クラスの検証セットの最初の画像で構成されています。かなり小さいが、オリジナルのImageNetデータセットの構造と多様性を保持している。

主な特徴

  • ImageNet10はImageNetのコンパクト版で、オリジナルデータセットの最初の10クラスを20枚の画像で表現している。
  • データセットはWordNetの階層に従って構成されており、ImageNetデータセットの構造を反映している。
  • CIテスト、サニティチェック、コンピュータビジョンタスクのトレーニングパイプラインの迅速なテストに最適です。
  • モデルのベンチマーク用に設計されたものではないが、モデルの基本的な機能と正しさを素早く示すことができる。

データセット構造

ImageNet10 データセットは、オリジナルの ImageNet と同様、WordNet の階層構造を用いて構成されている。ImageNet10 の 10 のクラスはそれぞれ、シンセット(同義語の集まり)によって記述される。ImageNet10 の画像は、1 つまたは複数の synset で注釈されており、さまざまなオブジェクトとその関係を認識するモデルをテストするためのコンパクトなリソースを提供します。

アプリケーション

ImageNet10データセットは、コンピュータビジョンモデルとパイプラインの迅速なテストとデバッグに役立ちます。サイズが小さいため、迅速な反復が可能で、継続的インテグレーションテストやサニティチェックに最適です。また、完全なImageNetデータセットを使った本格的なテストに移行する前に、新しいモデルの予備テストや既存モデルの変更を迅速に行うためにも使用できます。

使用方法

画像サイズ224x224のImageNet10データセットでディープラーニング・モデルをテストするには、以下のコード・スニペットを使用できます。利用可能な引数の包括的なリストについては、モデルのトレーニングページを参照してください。

テスト例

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n-cls.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="imagenet10", epochs=5, imgsz=224)
# Start training from a pretrained *.pt model
yolo train data=imagenet10 model=yolov8n-cls.pt epochs=5 imgsz=224

サンプル画像と注釈

ImageNet10データセットには、オリジナルのImageNetデータセットから画像のサブセットが含まれている。これらの画像は、データセットの最初の10クラスを表すように選択され、迅速なテストと評価のための多様かつコンパクトなデータセットを提供します。

データセットサンプル画像 この例では、ImageNet10データセットの画像の多様性と複雑さを紹介し、サニティチェックやコンピュータビジョンモデルの迅速なテストに有用であることを強調しています。

引用と謝辞

研究または開発で ImageNet10 データセットを使用する場合は、元の ImageNet 論文を引用してください:

@article{ILSVRC15,
         author = {Olga Russakovsky and Jia Deng and Hao Su and Jonathan Krause and Sanjeev Satheesh and Sean Ma and Zhiheng Huang and Andrej Karpathy and Aditya Khosla and Michael Bernstein and Alexander C. Berg and Li Fei-Fei},
         title={ImageNet Large Scale Visual Recognition Challenge},
         year={2015},
         journal={International Journal of Computer Vision (IJCV)},
         volume={115},
         number={3},
         pages={211-252}
}

我々は、Olga Russakovsky、Jia Deng、Li Fei-Feiが率いるImageNetチームがImageNetデータセットの作成と保守を行っていることに感謝したい。ImageNet10データセットは、コンパクトなサブセットではあるが、機械学習とコンピュータビジョンの研究コミュニティにおける迅速なテストとデバッグのための貴重なリソースである。ImageNetデータセットとその作成者の詳細については、ImageNetのウェブサイトをご覧ください。



Created 2023-11-12, Updated 2024-06-02
Authors: glenn-jocher (6)

コメント