WebMix-38M
收藏资源简介:
WebMix Parquet 是一个从多个公开图像-文本数据源整合而成的数据集,源数据集包括 cc12m、datacomp12m、yfcc15m 和 mscoco。数据以 Parquet 格式存储,图像以 Hugging Face `Image` 列格式(编码字节)保存。数据集包含多个结构化字段:核心字段为 `image`(图像)和 `caption`(文本描述);元数据字段包括 `source`(数据来源标识)、`split`(训练/验证分割)、`image_id`(图像唯一标识)、`source_dataset`(原始数据集名称)、`source_url`(原始图像URL)、`filename`(文件名)、`width`/`height`(图像尺寸)、`image_sha256`(图像哈希值)以及 `source_index`、`source_split`、`caption_index` 等索引字段。数据生成时,训练数据(`split` 为 train)来自 `cc12m`、`datacomp12m`、`yfcc15m` 三个源,而验证/支持数据(`split` 为 val)则来自 `mscoco` 源。该数据集适用于需要大规模图像-文本对的任务,如图像描述生成、视觉-语言预训练(VLP)、跨模态检索等。数据生成过程经过了详细配置,包括图像格式(JPEG)、质量、并行下载与处理参数,确保了数据的一致性和可处理性。
WebMix Parquet is a dataset integrated from multiple public image-text data sources, including cc12m, datacomp12m, yfcc15m, and mscoco. The data is stored in Parquet format, with images saved in the Hugging Face `Image` column format (encoded bytes). The dataset contains multiple structured fields: core fields are `image` (image) and `caption` (text description); metadata fields include `source` (data source identifier), `split` (train/validation split), `image_id` (unique image identifier), `source_dataset` (original dataset name), `source_url` (original image URL), `filename` (file name), `width`/`height` (image dimensions), `image_sha256` (image hash value), as well as index fields such as `source_index`, `source_split`, and `caption_index`. During data generation, training data (`split` as train) comes from the three sources `cc12m`, `datacomp12m`, and `yfcc15m`, while validation/support data (`split` as val) comes from the `mscoco` source. This dataset is suitable for tasks requiring large-scale image-text pairs, such as image caption generation, visual-language pre-training (VLP), and cross-modal retrieval. The data generation process is detailedly configured, including image format (JPEG), quality, parallel download, and processing parameters, ensuring data consistency and processability.
数据集概述:WebMix-38M
WebMix-38M 是一个图像-文本配对数据集,以 Parquet 格式存储,专为多模态模型训练设计。数据集主要包含来自多个公开来源的图像及其对应的文本描述(标题)。
数据规模与来源
- 总样本量:约3800万条(38M),来自以下四个公开数据集:
- CC12M
- DataComp 12M
- YFCC15M
- MS COCO(仅作为验证/支持集使用)
- 数据划分:
- 训练集:来自 CC12M、DataComp 12M、YFCC15M。
- 验证集:来自 MS COCO。
数据格式与存储
- 文件格式:Parquet,采用 ZSTD 压缩。
- 图像存储:图像以 JPEG 格式编码(质量90),以 Hugging Face 的
Image列格式存储为字节数据。 - 分片:每个分片包含约 100,000 行数据,便于分布式处理。
数据特征(字段)
每条记录包含以下字段:
| 字段名 | 数据类型 | 说明 |
|---|---|---|
| image | Image | 图像数据(字节编码) |
| caption | string | 图像的文本描述 |
| source | string | 原始来源数据集名称 |
| split | string | 数据划分(train/val) |
| image_id | string | 图像唯一标识符 |
| source_index | int64 | 来源数据集的索引 |
| source_dataset | string | 完整来源数据集名称 |
| source_split | string | 来源数据集中的划分 |
| source_url | string | 图像的原始下载链接 |
| filename | string | 文件名 |
| width | int32 | 图像宽度(像素) |
| height | int32 | 图像高度(像素) |
| image_sha256 | string | 图像的 SHA-256 哈希值 |
| caption_index | int32 | 标题在该图像下的索引 |
数据处理参数
- 最小图像尺寸:16 像素(小于此尺寸的图像被过滤)。
- 下载重试:最多 5 次下载重试,4 次 URL 重试。
- 并行处理:16 个进程用于数据生成,8 个进程用于下载。
- 日志间隔:每处理 10,000 条记录输出一次日志。
适用场景
- 多模态模型(如 CLIP、BLIP 等)的预训练或微调。
- 图像-文本对比学习、图像生成等任务。




