Набор данных 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.

Комментарии