OA0
OA0 是一个探索 AI 的社区
现在注册
已注册用户请  登录
说明
  • 条目同步来自 Hugging Face 主要方便针对性讨论。
OA0  ›  数据集  ›  PyTorch Image Models

timm/imagenet-1k-wds

 
  PyTorch Image Models  ·  2024-01-07 18:12:43  ·  23 次点击  ·  0 条评论  

数据集简介

这是一个基于 ImageNet-1K 数据集构建的 WebDataset 格式版本,由 timm 提供。它将经典的 ImageNet 图像分类数据集转换为更适合高效流式加载和分布式训练的格式。

主要用途

主要用于计算机视觉模型的训练与评估,特别是图像分类任务。其 WebDataset 格式设计旨在优化大规模数据集的加载效率,适用于需要高性能数据管道的深度学习训练场景。

数据类型/模态

  • 图像:来自 ImageNet-1K 的图像数据。
  • 文本:包含与图像对应的标签或类别信息。

规模与格式

  • 数据量:包含 80,700 个数据样本(行)。
  • 存储格式:采用 WebDataset 格式。这是一种将数据集存储为 TAR 文件集合的格式,每个文件包含图像、标签等数据对,便于流式读取,能有效减少 I/O 瓶颈。
  • 关联工具:与 datasetswebdatasetmlcroissant 等库兼容。

使用场景

  • 使用 PyTorch 等框架进行图像分类模型(如 ResNet、ViT 等)的训练和微调。
  • 需要快速迭代和高效数据加载的大规模实验或生产环境。
  • 作为基准数据集,用于模型性能的评估与比较。
23 次点击  ∙  0 人收藏  
登录后收藏  
0 条回复
关于 ·  帮助 ·  PING ·  隐私政策 ·  服务条款   
OA0 - Omni AI 0 一个探索 AI 的社区
沪ICP备2024103595号-2
耗时 14 ms
Developed with Cursor