Набор данных ImageNette#
Датасет ImageNette представляет собой подмножество ImageNet с 10 легко различимыми классами, созданное fast.ai, чтобы предоставить более быструю и удобную в использовании версию ImageNet для разработки программного обеспечения и обучения. Он содержит 13 394 цветных изображения (9469 для обучения и 3925 для валидации), охватывающих такие классы, как линь, английский спрингер, кассетный плеер и валторна, благодаря чему модели image classification обучаются за считанные минуты, а не за часы, требующиеся для полного набора данных ImageNet из 1000 классов.
Ключевые особенности#
- ImageNette содержит 13 394 изображения в 10 классах: линь, английский спрингер, кассетный магнитофон, бензопила, церковь, валторна, мусоровоз, бензоколонка, мяч для гольфа и парашют.
- Он поставляется с предопределенным разделением на 9 469 цветных обучающих и 3 925 проверочных изображений различных размеров.
- 10 хорошо различимых классов делают ImageNette удобным для быстрой тренировки и простого анализа, поэтому он широко используется для прототипирования и обучения классификации изображений.
Структура набора данных#
ImageNette поставляется с предопределенным разделением на обучение/проверку, при этом каждый класс хранится в своей отдельной папке:
| Split | Изображения | Классы |
|---|---|---|
| Обучение | 9,469 | 10 |
| Validation | 3,925 | 10 |
Точное количество изображений варьируется в зависимости от класса, обеспечивая реалистичное распределение для быстрого обучения и тестирования без масштабов полного набора данных ImageNet.
Применение#
Датасет ImageNette широко используется для обучения и оценки моделей deep learning в задачах классификации изображений, таких как Convolutional Neural Networks (CNN) и другие алгоритмы машинного обучения. Его простой формат и тщательно подобранные классы делают его удобным ресурсом как для начинающих, так и для опытных специалистов в области machine learning и computer vision.
Использование#
Чтобы обучить модель на датасете ImageNette в течение 100 epochs со стандартным размером изображения 224x224, ты можешь использовать следующие фрагменты кода. Полный список доступных аргументов см. на странице Training модели.
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="imagenette", epochs=100, imgsz=224)Примеры изображений и аннотации#
Датасет ImageNette содержит цветные изображения различных объектов и сцен, представляя собой разнообразный набор данных для задач image classification. Вот несколько примеров изображений из датасета:

Этот пример демонстрирует разнообразие и сложность изображений в наборе данных ImageNette, подчеркивая важность разнообразного набора данных для обучения надежных моделей классификации изображений.
ImageNette160 и ImageNette320#
Для более быстрого прототипирования и обучения ImageNette также доступен в двух уменьшенных размерах: ImageNette160 и ImageNette320. Эти датасеты сохраняют те же классы и структуру, что и полный ImageNette, но изображения в них изменены до меньшей максимальной длины стороны. Они особенно полезны для предварительного тестирования моделей или при ограниченности вычислительных ресурсов.
Чтобы использовать эти датасеты, просто замени imagenette на imagenette160 или imagenette320 в команде обучения. Следующие фрагменты кода иллюстрируют это:
from ultralytics import YOLO
# Load a model
model = YOLO("yolo26n-cls.pt") # load a pretrained model (recommended for training)
# Train the model with ImageNette160
results = model.train(data="imagenette160", epochs=100, imgsz=160)from ultralytics import YOLO
# Load a model
model = YOLO("yolo26n-cls.pt") # load a pretrained model (recommended for training)
# Train the model with ImageNette320
results = model.train(data="imagenette320", epochs=100, imgsz=320)Эти уменьшенные версии датасета позволяют быстро выполнять итерации в процессе разработки, обеспечивая при этом реалистичные задачи классификации изображений. Ты также можешь управлять датасетами классификации и запускать обучение в облаке с помощью Ultralytics Platform.
Цитирование и благодарности#
Если ты используешь датасет ImageNette в своих исследованиях или разработке, пожалуйста, укажи соответствующую ссылку. Для получения дополнительной информации об ImageNette посети страницу датасета ImageNette на GitHub.
Мы хотели бы выразить признательность команде fast.ai за создание и поддержку ImageNette в качестве ценного ресурса для исследовательского сообщества machine learning и computer vision.
FAQ#
Датасет ImageNette представляет собой упрощенное подмножество ImageNet с 10 легко различимыми классами, такими как линь, английский спрингер и валторна. Он содержит 13 394 изображения (9469 для обучения и 3925 для валидации) и был создан fast.ai, чтобы предложить более управляемый набор данных для эффективного обучения и оценки моделей классификации изображений. Это делает его особенно полезным для быстрой разработки программного обеспечения и образовательных целей в области machine learning и компьютерного зрения.
ImageNette содержит 13 394 изображения в общей сложности — 9 469 для обучения и 3 925 для проверки — по 10 классам: линь, английский спрингер, кассетный магнитофон, бензопила, церковь, валторна, мусоровоз, бензоколонка, мяч для гольфа и парашют. Каждый класс хранится в своей собственной папке в соответствии со стандартной структурой классификации, которую ожидает Ultralytics.
Чтобы обучить модель YOLO на ImageNette в течение 100 epochs, используй следующие команды. Убедись, что у тебя настроено окружение 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="imagenette", epochs=100, imgsz=224)Более подробную информацию см. на странице документации Training.
Набор данных ImageNette выгоден по нескольким причинам:
- Быстро и просто: Имея всего 10 классов и около 13 000 изображений, он гораздо менее сложен и требует меньше времени на обучение, чем полный ImageNet.
- Образовательное использование: Идеально подходит для изучения и преподавания основ классификации изображений, так как требует меньше вычислительной мощности и времени.
- Универсальность: Широко используется для обучения и оценки моделей классификации изображений, особенно Convolutional Neural Networks (CNN).
Для получения более подробной информации об обучении моделей и управлении датасетами изучи раздел Dataset Structure.
Да, ImageNette также доступен в двух измененных версиях, ImageNette160 и ImageNette320, изображения в которых уменьшены до меньшей максимальной длины края. Эти версии помогают при быстром прототипировании и особенно полезны, когда вычислительные ресурсы ограничены.
Пример обучения с ImageNette160from ultralytics import YOLO # Load a model model = YOLO("yolo26n-cls.pt") # Train the model with ImageNette160 results = model.train(data="imagenette160", epochs=100, imgsz=160)Для получения дополнительной информации обратись к разделу Training with ImageNette160 and ImageNette320.
Набор данных ImageNette широко используется в:
- Образовательные цели: Для обучения новичков основам машинного обучения и computer vision.
- Разработке программного обеспечения: Для быстрого прототипирования и разработки моделей классификации изображений.
- Исследования в области глубокого обучения: Для оценки и тестирования производительности различных моделей глубокого обучения, особенно сверточных Neural Networks (CNN).
Изучи раздел Applications для получения подробных сценариев использования.



