Перейти к содержимому

Набор данных Fashion-MNIST

Fashion-MNIST dataset - это база данных изображений товаров Zalando, состоящая из обучающего набора из 60 000 примеров и тестового набора из 10 000 примеров. Каждый пример - это изображение размером 28х28 градаций серого, связанное с меткой из 10 классов. Fashion-MNIST призван служить прямой заменой оригинальному набору данных MNIST для бенчмаркинга алгоритмов машинного обучения.

Основные характеристики

  • Fashion-MNIST содержит 60 000 обучающих и 10 000 тестирующих изображений артикулов Zalando.
  • Набор данных состоит из полутоновых изображений размером 28х28 пикселей.
  • С каждым пикселем связано одно пиксельное значение, указывающее на светлоту или темноту этого пикселя, причем более высокие числа означают более темный цвет. Это пиксельное значение - целое число от 0 до 255.
  • Мод-MNIST широко используется для обучения и тестирования в области машинного обучения, особенно для задач классификации изображений.

Структура набора данных

Набор данных Fashion-MNIST разбит на два подмножества:

  1. Набор для обучения: Этот поднабор содержит 60 000 изображений, используемых для обучения моделей машинного обучения.
  2. Набор для тестирования: Это подмножество состоит из 10 000 изображений, которые используются для тестирования и бенчмаркинга обученных моделей.

Ярлыки

Каждому тренировочному и тестовому примеру присваивается одна из следующих меток:

  1. Футболка/топ
  2. Брюки
  3. Пуловер
  4. Платье
  5. Пальто
  6. Сандал
  7. Рубашка
  8. Кроссовки
  9. Сумка
  10. Ботинок по щиколотку

Приложения

Набор данных Fashion-MNIST широко используется для обучения и оценки моделей глубокого обучения в задачах классификации изображений, таких как конволюционные нейронные сети (CNN), машины опорных векторов (SVM) и различные другие алгоритмы машинного обучения. Простой и хорошо структурированный формат набора данных делает его незаменимым ресурсом для исследователей и практиков в области машинного обучения и компьютерного зрения.

Использование

Чтобы обучить модель CNN на наборе данных Fashion-MNIST в течение 100 эпох при размере изображения 28х28, ты можешь использовать следующие фрагменты кода. Полный список доступных аргументов ты найдешь на странице обучения модели.

Пример поезда

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="fashion-mnist", epochs=100, imgsz=28)
# Start training from a pretrained *.pt model
yolo detect train data=fashion-mnist model=yolov8n-cls.pt epochs=100 imgsz=28

Примеры изображений и аннотаций

Набор данных Fashion-MNIST содержит полутоновые изображения артикулов Zalando, представляя собой хорошо структурированный набор данных для задач классификации изображений. Вот несколько примеров изображений из этого набора:

Образец изображения из набора данных

Этот пример демонстрирует разнообразие и сложность изображений в наборе данных Fashion-MNIST, подчеркивая важность разнообразных наборов данных для обучения надежных моделей классификации изображений.

Благодарности

Если ты используешь набор данных Fashion-MNIST в своих исследованиях или разработках, пожалуйста, отметь его, поставив ссылку на репозиторий GitHub. Этот набор данных был предоставлен компанией Zalando Research.



Created 2023-11-12, Updated 2024-06-02
Authors: glenn-jocher (4)

Комментарии