12 KiB
comments | description | keywords |
---|---|---|
true | Ultralyticsがサポートするオブジェクト検出、セグメンテーション、ポーズ推定、画像分類、マルチオブジェクト追跡に対応した各種コンピュータビジョンデータセットを探求します。 | コンピュータビジョン, データセット, Ultralytics, YOLO, オブジェクト検出, インスタンスセグメンテーション, ポーズ推定, 画像分類, マルチオブジェクト追跡 |
データセット概要
Ultralyticsは、検出、インスタンスセグメンテーション、ポーズ推定、分類、マルチオブジェクト追跡などのコンピュータビジョンタスクを容易にするために、様々なデータセットをサポートしています。以下は主要なUltralyticsデータセットのリストであり、それに続いて各コンピュータビジョンタスクの概要と関連するデータセットが記載されています。
!!! note
🚧 当社の多言語ドキュメントは現在作成中であり、改善のため頑張って作業しています。ご理解いただいてありがとうございます! 🙏
検出データセット
バウンディングボックスオブジェクト検出は、画像内のオブジェクトを検出し、各オブジェクトの周りにバウンディングボックスを描くことによって局在化するコンピュータビジョン技術です。
- Argoverse: 都市環境からの3D追跡と運動予測データを含むデータセットで、詳細なアノテーションが付けられています。
- COCO: オブジェクト検出、セグメンテーション、キャプショニング用に設計された大規模データセットで、20万以上のラベル付き画像が含まれています。
- COCO8: COCO trainとCOCO valの最初の4画像を含み、クイックテストに適しています。
- Global Wheat 2020: 世界中から収集された小麦頭画像のデータセットで、オブジェクト検出と局在化タスク用です。
- Objects365: 365のオブジェクトカテゴリと60万以上のアノテート画像を含む高品質の大規模オブジェクト検出データセットです。
- OpenImagesV7: Googleによる総合的なデータセットで、170万のトレイン画像と42kのバリデーション画像を含んでいます。
- SKU-110K: 小売環境での密集したオブジェクト検出を特徴とするデータセットで、11Kの画像と170万のバウンディングボックスがあります。
- VisDrone: ドローンがキャプチャした映像からのオブジェクト検出とマルチオブジェクト追跡データを含むデータセットで、1万以上の画像とビデオシーケンスがあります。
- VOC: パスカルビジュアルオブジェクトクラス(VOC)データセットで、オブジェクト検出とセグメンテーション用に、20のオブジェクトクラスと11K以上の画像が含まれています。
- xView: 上空画像でのオブジェクト検出用データセットで、60のオブジェクトカテゴリを含み、100万以上のアノテートオブジェクトが含まれています。
インスタンスセグメンテーションデータセット
インスタンスセグメンテーションは、画像内のオブジェクトをピクセルレベルで識別し、局在化するコンピュータビジョン技術です。
- COCO: オブジェクト検出、セグメンテーション、キャプショニングタスク用に設計された大規模データセットで、20万以上のラベル付き画像が含まれています。
- COCO8-seg: セグメンテーションアノテーションを含む8つのCOCO画像を含んだ、より小さなインスタンスセグメンテーションタスク用データセットです。
ポーズ推定
ポーズ推定は、カメラや世界座標系に対するオブジェクトのポーズを決定するために使用される技術です。
- COCO: ポーズ推定タスク用に設計された大規模データセットで、人間のポーズアノテーションが含まれています。
- COCO8-pose: 人間のポーズアノテーションを含む8つのCOCO画像を含んだ、小規模のポーズ推定タスク用データセットです。
- Tiger-pose: タイガーのポーズ推定タスク用に、263枚の画像で構成され、1匹のタイガーに対して12個のキーポイントがアノテートされたコンパクトなデータセットです。
分類
画像分類は、そのビジュアルコンテンツに基づいて一つまたは複数の定義済みクラスやカテゴリに画像を分類するコンピュータビジョンタスクです。
- Caltech 101: 101のオブジェクトカテゴリを含む画像で構成された、画像分類タスク用データセッ トです。
- Caltech 256: Caltech 101の拡張版で、256のオブジェクトカテゴリとより難易度の高い画像が含まれています。
- CIFAR-10: 10クラスに60Kの32x32カラー画像が含まれるデータセットで、クラスごとに6Kの画像があります。
- CIFAR-100: CIFAR-10の拡張版で、100のオブジェクトカテゴリとクラスごとに600の画像が含まれています。
- Fashion-MNIST: 画像分類タスク用の70,000枚のグレースケールのファッションカテゴリ画像で構成されるデータセットです。
- ImageNet: オブジェクト検出および画像分類用の大規模データセットで、1400万枚を超える画像と20,000カテゴリが含まれています。
- ImageNet-10: ImageNetのより小さなサブセットで、高速な実験とテストに適した10カテゴリが含まれています。
- Imagenette: ImageNetから抽出された10個の簡単に識別可能なクラスを含む、より小さなサブセットです。
- Imagewoof: ImageNetから抽出された10の犬種カテゴリを含む、画像分類タスクにさらに挑戦的なサブセットです。
- MNIST: 手書き数字の70,000枚のグレースケール画像で構成される画像分類タスク用データセットです。
指向性バウンディングボックス
指向性バウンディングボックス(OBB)は、回転したバウンディングボックスを使用して画像内の斜めのオブジェクトを検出するためのコンピュータビジョン手法で、主に航空写真や衛星画像へ適用されます。
- DOTAv2: 170万インスタンスと11,268枚の画像を含む人気のOBB航空画像データセットです。
マルチオブジェクト追跡
マルチオブジェクト追跡は、複数オブジェクトをビデオシーケンスで時間を追って検出し、追跡するコンピュータビジョン技術です。
- Argoverse: 都市環境からの3D追跡と運動予測データを含むデータセットで、マルチオブジェクト追跡タスクに富んだアノテーションがあります。
- VisDrone: ドローンがキャプチャした映像からのオブジェクト検出とマルチオブジェクト追跡データを含むデータセットで、1万以上の画像とビデオシーケンスがあります。
新しいデータセットの提供
新しいデータセットを提供するには、既存のインフラと適合するように複数のステップが必要です。以下が必要なステップです:
新しいデータセットを提供する手順
-
画像を収集します:データセットに属する画像を収集します。これらは、公共データベースや自分のコレクションなど、さまざまなソースから収集することができます。
-
画像をアノテートします:これらの画像を、タスクに応じてバウンディングボックス、セグメント、またはキーポイントでアノテートします。
-
アノテーションをエクスポートします:これらのアノテーションを、UltralyticsがサポートしているYOLO *.txtファイルフォーマットに変換します。
-
データセットを編成します:データセットを正しいフォルダ構造に配置します。
train/
およびval/
のトップレベルディレクトリーを持ち、各ディレクトリー内にimages/
およびlabels/
のサブディレクトリーがあるべきです。dataset/ ├── train/ │ ├── images/ │ └── labels/ └── val/ ├── images/ └── labels/
-
data.yaml
ファイルを作成します:データセットのルートディレクトリに、データセット、クラス、その他の必要な情報を記述するdata.yaml
ファイルを作成します。 -
**画像を最適化(オプショナル)**します:データセットのサイズを削減してより効率的な処理を行いたい場合は、以下のコードを使用して画像を最適化することができます。これは必須ではありませんが、データセットのサイズを小さくし、ダウンロード速度を上げるために推奨されます。
-
データセットをZipします:データセットフォルダ全体をzipファイルに圧縮します。
-
ドキュメントとPRを作成します:データセットに関するドキュメンテーションページを作成し、既存のフレームワークにどのように適合するかを説明します。その後、Pull Request (PR)を提出します。PRを提出する方法の詳細については、Ultralyticsの貢献ガイドラインを参照してください。
データセットを最適化してZipするためのサンプルコード
!!! example "データセットを最適化してZipする"
=== "Python"
```python
from pathlib import Path
from ultralytics.data.utils import compress_one_image
from ultralytics.utils.downloads import zip_directory
# データセットディレクトリを定義
path = Path('path/to/dataset')
# データセット内の画像を最適化 (オプショナル)
for f in path.rglob('*.jpg'):
compress_one_image(f)
# 'path/to/dataset.zip'にデータセットをZip
zip_directory(path)
```
これらのステップに従えば、Ultralyticsの既存の構造に適切に統合された新しいデータセットを提供することができます。