Link to this sectionНабор данных Dog-Pose#
Link to this sectionВведение#
Набор данных Ultralytics Dog-Pose — это высококачественный и обширный датасет, специально подготовленный для оценки ключевых точек у собак, содержащий 6 773 обучающих и 1 703 проверочных изображения.
Watch: How to Train an Ultralytics YOLO Model on the Stanford Dog Pose Estimation Dataset | Step-by-Step Tutorial
Каждое аннотированное изображение включает 24 ключевые точки с 3 измерениями для каждой (x, y, видимость), что делает его ценным ресурсом для передовых исследований и разработок в области компьютерного зрения.
Для конкретной породы или другого животного Ultralytics Platform позволяет загружать, размечать и обучать пользовательскую модель ключевых точек на твоих собственных данных без необходимости управления инфраструктурой.
Link to this sectionСтруктура набора данных#
-
Всего изображений: 8 476 (6 773 для обучения / 1 703 для проверки) с соответствующими файлами аннотаций в формате YOLO.
-
Ключевые точки: 24 для каждой собаки с триплетами
(x, y, visibility). -
Размер загрузки: ~337 МБ.
-
Структура:
datasets/dog-pose/ ├── images/{train,val} └── labels/{train,val}
Link to this sectionYAML набора данных#
Для настройки датасета используется YAML-файл. Он включает пути, сведения о ключевых точках и другую релевантную информацию. Файл dog-pose.yaml для датасета Dog-Pose доступен по ссылке https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/dog-pose.yaml.
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# Dogs dataset http://vision.stanford.edu/aditya86/ImageNetDogs/ by Stanford
# Documentation: https://docs.ultralytics.com/datasets/pose/dog-pose
# Example usage: yolo train data=dog-pose.yaml
# parent
# ├── ultralytics
# └── datasets
# └── dog-pose ← downloads here (337 MB)
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: dog-pose # dataset root dir
train: images/train # train images (relative to 'path') 6773 images
val: images/val # val images (relative to 'path') 1703 images
# Keypoints
kpt_shape: [24, 3] # number of keypoints, number of dims (2 for x,y or 3 for x,y,visible)
# Classes
names:
0: dog
# Keypoint names per class
kpt_names:
0:
- front_left_paw
- front_left_knee
- front_left_elbow
- rear_left_paw
- rear_left_knee
- rear_left_elbow
- front_right_paw
- front_right_knee
- front_right_elbow
- rear_right_paw
- rear_right_knee
- rear_right_elbow
- tail_start
- tail_end
- left_ear_base
- right_ear_base
- nose
- chin
- left_ear_tip
- right_ear_tip
- left_eye
- right_eye
- withers
- throat
# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/dog-pose.zipLink to this sectionИспользование#
Чтобы обучить модель YOLO26n-pose на наборе данных Dog-Pose в течение 100 эпох с размером изображения 640, ты можешь использовать следующие фрагменты кода. Полный список доступных аргументов см. на странице Обучение модели.
from ultralytics import YOLO
# Load a model
model = YOLO("yolo26n-pose.pt") # load a pretrained model (recommended for training)
# Train the model
results = model.train(data="dog-pose.yaml", epochs=100, imgsz=640)Link to this sectionПримеры изображений и аннотации#
Вот несколько примеров изображений из набора данных Dog-Pose вместе с соответствующими аннотациями:
- Мозаичное изображение: это изображение демонстрирует обучающий пакет, составленный из мозаично собранных изображений. Мозаика — это техника, используемая во время обучения, которая объединяет несколько изображений в одно, чтобы увеличить разнообразие объектов и сцен в каждом обучающем пакете. Это помогает улучшить способность модели к обобщению при работе с объектами различных размеров, соотношений сторон и контекстов.
Этот пример демонстрирует разнообразие и сложность изображений в наборе данных Dog-Pose, а также преимущества использования мозаики в процессе обучения.
Link to this sectionЦитирование и благодарности#
Если ты используешь набор данных Dog-Pose в своей исследовательской или проектной работе, пожалуйста, сошлися на следующую статью:
@inproceedings{khosla2011fgvc,
title={Novel dataset for Fine-Grained Image Categorization},
author={Aditya Khosla and Nityananda Jayadevaprakash and Bangpeng Yao and Li Fei-Fei},
booktitle={First Workshop on Fine-Grained Visual Categorization (FGVC), IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
year={2011}
}
@inproceedings{deng2009imagenet,
title={ImageNet: A Large-Scale Hierarchical Image Database},
author={Jia Deng and Wei Dong and Richard Socher and Li-Jia Li and Kai Li and Li Fei-Fei},
booktitle={IEEE Computer Vision and Pattern Recognition (CVPR)},
year={2009}
}Мы хотим поблагодарить команду Стэнфорда за создание и поддержку этого ценного ресурса для сообщества компьютерного зрения. Для получения дополнительной информации о наборе данных Dog-Pose и его создателях посети сайт набора данных Stanford Dogs.
Link to this sectionFAQ#
Link to this sectionЧто такое набор данных Dog-Pose и как он используется с Ultralytics YOLO26?#
Набор данных Dog-Pose содержит 6 773 обучающих и 1 703 проверочных изображения, размеченных 24 ключевыми точками для оценки позы собаки. Он предназначен для обучения и проверки моделей с помощью Ultralytics YOLO26 и поддерживает такие задачи, как анализ поведения животных, мониторинг питомцев и ветеринарные исследования. Комплексные аннотации делают этот набор данных идеальным для разработки точных моделей оценки позы собак.
Link to this sectionКак мне обучить модель YOLO26 с использованием набора данных Dog-Pose в Ultralytics?#
Загрузи yolo26n-pose.pt и вызови model.train(data="dog-pose.yaml", epochs=100, imgsz=640) — см. Пример обучения выше для получения полных фрагментов кода Python и CLI, а также страницу Обучение для получения полного списка аргументов.
Link to this sectionКаковы преимущества использования набора данных Dog-Pose?#
Благодаря 8 476 изображениям в сумме (6 773 для обучения / 1 703 для проверки), охватывающим широкий спектр пород и поз собак, а также 24 ключевым точкам в 3 измерениях (x, y, видимость) для каждой аннотации, набор данных Dog-Pose обеспечивает покрытие реальных сценариев, необходимых для таких приложений, как мониторинг питомцев и анализ поведения. Подробнее о характеристиках и использовании см. в разделе Введение в набор данных.
Link to this sectionКак мозаика помогает процессу обучения YOLO26 при использовании набора данных Dog-Pose?#
Мозаика объединяет несколько изображений из Dog-Pose в одно обучающее изображение, увеличивая разнообразие поз, размеров и фонов, которые модель видит на каждом шаге. Это улучшает обобщающую способность в новых контекстах и масштабах, одновременно снижая переобучение. Примеры изображений см. в разделе Примеры изображений и аннотаций.
Link to this sectionГде я могу найти YAML-файл набора данных Dog-Pose и как его использовать?#
YAML-файл набора данных Dog-Pose можно найти по адресу https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/dog-pose.yaml. Этот файл определяет конфигурацию датасета, включая пути, классы, детали ключевых точек и другую релевантную информацию. YAML указывает 24 ключевые точки с 3 измерениями для каждой, что делает его пригодным для детальных задач оценки позы.
Чтобы использовать этот файл со скриптами обучения YOLO26, просто укажи его в своей команде обучения, как показано в разделе Использование. Набор данных будет автоматически загружен при первом использовании, что делает настройку очень простой.
Подробнее о моделях ключевых точек читай в документации к задаче Оценка позы.