Ultralytics YOLO27:
No license

CIFAR-10 数据集#

CIFAR-10(加拿大高等研究院)数据集是一个经典的图像分类基准,包含 60,000 张 32x32 彩色图像,均匀分为 10 个类别——飞机、汽车、鸟、猫、鹿、狗、青蛙、马、船和卡车。该数据集采用预定义的数据划分,包括 50,000 张训练图像和 10,000 张测试图像(每类 6,000 张),是训练和基准测试分类模型的轻量级且均衡的起点。如需更细粒度的挑战,请参阅相关的 CIFAR-100 数据集。

访问 Ultralytics Platform 上的 CIFAR-10,预览样本、检查数据集统计信息,并克隆数据集用于训练。



Watch: How to Train an Image Classification Model with CIFAR-10 Dataset using Ultralytics YOLO

主要特性#

  • CIFAR-10 包含 60,000 张 32x32 像素的彩色图像,均匀分为 10 个类别。
  • 每个类别恰好包含 6,000 张图像——其中 5,000 张用于训练,1,000 张用于测试——因此该数据集完全均衡。
  • 这 10 个类别分别是飞机、汽车、鸟、猫、鹿、狗、青蛙、马、船和卡车。
  • 该数据集提供预定义的训练集/测试集划分,因此无需手动或自动进行划分。
  • CIFAR-10 是图像分类和目标识别研究的标准基准。

数据集结构#

CIFAR-10 自带官方预定义划分,因此无需自动或手动进行分区:

  • 类别:10(飞机、汽车、鸟、猫、鹿、狗、青蛙、马、船、卡车)
  • 图像总数:60,000 张(32x32 彩色图像)
  • 训练集:50,000 张图像(每类 5,000 张)
  • 测试集:10,000 张图像(每类 1,000 张)
验证集划分

CIFAR-10 没有单独的验证文件夹,因此 Ultralytics 默认在训练期间将包含 10,000 张图像的测试集用作验证集划分。

应用#

CIFAR-10 被广泛用于训练和评估图像分类模型,涵盖经典的卷积神经网络(CNNs)和支持向量机(SVMs),以及现代深度架构。其较小的图像尺寸和均衡的类别使其非常适合快速实验、新算法基准测试,以及教授计算机视觉基础知识。

使用方法#

在 CIFAR-10 上训练 YOLO 模型 100 个周期,图像尺寸设为 32。如需查看所有可用参数,请参阅训练页面和图像分类任务指南。

训练示例
from ultralytics import YOLO

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

# Train the model
results = model.train(data="cifar10", epochs=100, imgsz=32)

示例图像和标注#

CIFAR-10 数据集包含各种物体的彩色图像,为图像分类任务提供了结构良好的数据集。以下是该数据集中的一些图像示例:

CIFAR-10 图像分类数据集样本

这些样本展示了 CIFAR-10 数据集中物体的多样性,突出了多样化数据集对于训练稳健的图像分类模型的价值。

引用和致谢#

如果你在研究或开发工作中使用 CIFAR-10 数据集,请引用以下论文:

引用
@TECHREPORT{Krizhevsky09learningmultiple,
            author={Alex Krizhevsky},
            title={Learning multiple layers of features from tiny images},
            institution={},
            year={2009}
}

我们感谢 Alex Krizhevsky 创建并维护 CIFAR-10 数据集,为机器学习计算机视觉研究社区提供了宝贵资源。如需了解有关 CIFAR-10 数据集及其创建者的更多信息,请访问 CIFAR-10 数据集网站

常见问题#

  • CIFAR-10 数据集被广泛用于训练和基准测试图像分类及目标识别模型。它包含 60,000 张 32x32 彩色图像,均匀分布在 10 个类别中;其较小的规模和均衡的类别使其成为卷积神经网络(CNNs)和支持向量机(SVMs)等算法的快速、可靠基准。

  • 要在 CIFAR-10 上训练 Ultralytics YOLO 模型,请使用下面的代码片段。数据集将在首次使用时自动下载。如需查看完整的参数列表,请参阅模型的训练页面。

    训练示例
    from ultralytics import YOLO
    
    # Load a model
    model = YOLO("yolo26n-cls.pt")  # load a pretrained model (recommended for training)
    
    # Train the model
    results = model.train(data="cifar10", epochs=100, imgsz=32)
  • CIFAR-10 有 10 个类别——飞机、汽车、鸟、猫、鹿、狗、青蛙、马、船和卡车——每类恰好包含 6,000 张图像,总计 60,000 张图像。这些类别相互排斥且完全均衡,类别之间没有重叠。

  • CIFAR-10 自带预定义划分,包括 50,000 张训练图像和 10,000 张测试图像,其中每类恰好包含 5,000 张训练图像和 1,000 张测试图像。不同于由 Ultralytics 自动划分的基于文件夹的分类数据集,CIFAR-10 的官方分区会原样使用,默认在训练期间将测试集用作验证集划分。

  • 可以。Ultralytics Platform 可以帮助你管理数据集、训练图像分类模型,并在无需大量编码的情况下部署模型。它是在云端运行 CIFAR-10 实验的便捷方式,你还可以在我们的分类数据集概览中探索更多选项。

评论