laion400m,laion400m

  • 説明:

LAION-400M データセットは完全にオープンで、自由にアクセスできます。

このデータセットの完全な説明については、 https://laion.ai/laion-400-open-dataset/を確認してください。

LAION-400M データセット内のすべての画像とテキストは、OpenAI の CLIP を使用して、テキストと画像の埋め込み間のコサイン類似度を計算し、類似度が 0.3 未満のものを削除することでフィルタリングされています。 0.3 というしきい値は、人間の評価によって決定されており、セマンティックな画像、テキスト、コンテンツの一致を推定するための優れたヒューリスティックであると思われました。

画像とテキストのペアは、Common Crawl Web データ ダンプから抽出されたもので、2014 年から 2021 年の間にクロールされたランダムな Web ページからのものです。

スプリット
@article{DBLP:journals/corr/abs-2111-02114,
  author    = {Christoph Schuhmann and
               Richard Vencu and
               Romain Beaumont and
               Robert Kaczmarczyk and
               Clayton Mullis and
               Aarush Katta and
               Theo Coombes and
               Jenia Jitsev and
               Aran Komatsuzaki},
  title     = { {LAION-400M:} Open Dataset of CLIP-Filtered 400 Million Image-Text
               Pairs},
  journal   = {CoRR},
  volume    = {abs/2111.02114},
  year      = {2021},
  url       = {https://arxiv.org/abs/2111.02114},
  eprinttype = {arXiv},
  eprint    = {2111.02114},
  timestamp = {Fri, 05 Nov 2021 15:25:54 +0100},
  biburl    = {https://dblp.org/rec/journals/corr/abs-2111-02114.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}

laion400m/images (デフォルト設定)

  • 機能構造:
FeaturesDict({
    'caption': Text(shape=(), dtype=string),
    'image': Image(shape=(None, None, 3), dtype=uint8),
    'license': Text(shape=(), dtype=string),
    'nsfw': ClassLabel(shape=(), dtype=int64, num_classes=4),
    'original_height': Scalar(shape=(), dtype=int32),
    'original_width': Scalar(shape=(), dtype=int32),
    'similarity': Scalar(shape=(), dtype=float64),
    'url': Text(shape=(), dtype=string),
})
  • 機能のドキュメント:
特徴クラスDtype説明値の範囲
特徴辞書
キャプション文章HTML 代替テキスト属性
画像画像(なし、なし、3) uint8画像
ライセンス文章クリエイティブ コモンズ ライセンスの種類 (該当する場合)
nsfwクラスラベルint64 NSFWタグ(CLIPで検出)。まとまりのないタグや欠落したタグは UNTAGGED に置き換えられます
オリジナルの高さスカラーint32画像の元の高さ
original_widthスカラーint32画像の元の幅
類似性スカラーfloat64テキストと画像の埋め込み間のコサイン類似度スコア。欠損値のデフォルトは -1.0 [0.0、1.0]
URL文章画像の URL

laion400m/embeddings

  • 機能構造:
FeaturesDict({
    'caption': Text(shape=(), dtype=string),
    'image_embedding': Tensor(shape=(512,), dtype=float16),
    'license': Text(shape=(), dtype=string),
    'nsfw': ClassLabel(shape=(), dtype=int64, num_classes=4),
    'original_height': Scalar(shape=(), dtype=int32),
    'original_width': Scalar(shape=(), dtype=int32),
    'similarity': Scalar(shape=(), dtype=float64),
    'text_embedding': Tensor(shape=(512,), dtype=float16),
    'url': Text(shape=(), dtype=string),
})
  • 機能のドキュメント:
特徴クラスDtype説明値の範囲
特徴辞書
キャプション文章HTML 代替テキスト属性
image_embeddingテンソル(512) float16 CLIP画像埋め込み
ライセンス文章クリエイティブ コモンズ ライセンスの種類 (該当する場合)
nsfwクラスラベルint64 NSFWタグ(CLIPで検出)。まとまりのないタグや欠落したタグは UNTAGGED に置き換えられます
オリジナルの高さスカラーint32画像の元の高さ
original_widthスカラーint32画像の元の幅
類似性スカラーfloat64テキストと画像の埋め込み間のコサイン類似度スコア。欠損値のデフォルトは -1.0 [0.0、1.0]
text_embeddingテンソル(512) float16 CLIP テキスト埋め込み
URL文章画像の URL