跳至内容

CIFAR-10 数据集

CIFAR-10(加拿大高级研究所)数据集是一个广泛用于机器学习和计算机视觉算法的图像集合。它由 CIFAR 研究所的研究人员开发,包含 10 个不同类别的 60,000 张 32x32 彩色图像。

主要功能

  • CIFAR-10 数据集由 60,000 张图像组成,分为 10 个等级。
  • 每个类别包含 6000 张图像,其中 5000 张用于训练,1000 张用于测试。
  • 图像是彩色的,大小为 32x32 像素。
  • 10 个不同的类别分别代表飞机、汽车、鸟、猫、鹿、狗、青蛙、马、船和卡车。
  • CIFAR-10 通常用于机器学习和计算机视觉领域的训练和测试。

数据集结构

CIFAR-10 数据集分为两个子集:

  1. 训练集:该子集包含 50,000 张图像,用于训练机器学习模型。
  2. 测试集:该子集由 10,000 张图像组成,用于测试和基准测试训练有素的模型。

应用

CIFAR-10 数据集广泛用于训练和评估图像分类任务中的深度学习模型,如卷积神经网络(CNN)、支持向量机(SVM)和其他各种机器学习算法。该数据集在类别方面的多样性和彩色图像的存在使其成为机器学习和计算机视觉领域研究和开发的全面数据集。

使用方法

要在图像大小为 32x32 的 CIFAR-10 数据集上训练YOLO 模型 100 次,可以使用以下代码片段。有关可用参数的完整列表,请参阅模型训练页面。

列车示例

from ultralytics import YOLO

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

# Train the model
results = model.train(data='cifar10', epochs=100, imgsz=32)
# Start training from a pretrained *.pt model
yolo detect train data=cifar10 model=yolov8n-cls.pt epochs=100 imgsz=32

图片和注释示例

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 数据集网站



创建于 2023-11-12,更新于 2023-11-22
作者:glenn-jocher(3)

评论