Conjunto de datos Dog-Pose#
Introducción#
El conjunto de datos Dog-Pose de Ultralytics es un conjunto de datos extenso y de alta calidad creado específicamente para la estimación de puntos clave en perros, que proporciona 6773 imágenes de entrenamiento y 1703 de validación.
Watch: How to Train an Ultralytics YOLO Model on the Stanford Dog Pose Estimation Dataset | Step-by-Step Tutorial
Cada imagen anotada incluye 24 puntos clave con 3 dimensiones por punto clave (x, y, visibilidad), lo que lo convierte en un recurso valioso para la investigación y el desarrollo avanzados en visión artificial.
Para una raza específica o un animal completamente diferente, Ultralytics Platform se encarga de subir, etiquetar y entrenar un modelo de puntos clave personalizado con tus propios datos sin necesidad de gestionar infraestructura.
Estructura del dataset#
-
Imágenes totales: 8.476 (6.773 de entrenamiento / 1.703 de validación) con archivos de etiquetas en formato YOLO correspondientes.
-
Puntos clave: 24 por perro con tripletes
(x, y, visibility). -
Tamaño de descarga: ~337 MB.
-
Diseño:
datasets/dog-pose/ ├── images/{train,val} └── labels/{train,val}
YAML del dataset#
Se utiliza un archivo YAML para definir la configuración del conjunto de datos. Incluye rutas, detalles de los puntos clave y otra información relevante. En el caso del conjunto de datos Dog-Pose, el archivo dog-pose.yaml está disponible en 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.zipUso#
Para entrenar un modelo YOLO26n-pose en el conjunto de datos Dog-Pose durante 100 épocas con un tamaño de imagen de 640, puedes usar los siguientes fragmentos de código. Para obtener una lista completa de los argumentos disponibles, consulta la página de Entrenamiento del modelo.
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)Muestras de imágenes y anotaciones#
Aquí tienes algunos ejemplos de imágenes del conjunto de datos Dog-Pose, junto con sus anotaciones correspondientes:
- Imagen en mosaico: Esta imagen muestra un lote de entrenamiento compuesto por imágenes del conjunto de datos en mosaico. El mosaico es una técnica utilizada durante el entrenamiento que combina múltiples imágenes en una sola para aumentar la variedad de objetos y escenas dentro de cada lote de entrenamiento. Esto ayuda a mejorar la capacidad del modelo para generalizar a diferentes tamaños de objeto, relaciones de aspecto y contextos.
El ejemplo muestra la variedad y complejidad de las imágenes en el conjunto de datos Dog-Pose y los beneficios de usar mosaicing durante el proceso de entrenamiento.
Citas y agradecimientos#
Si utilizas el conjunto de datos Dog-Pose en tu trabajo de investigación o desarrollo, por favor cita el siguiente artículo:
@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}
}Queremos agradecer al equipo de Stanford por crear y mantener este valioso recurso para la comunidad de visión artificial. Para obtener más información sobre el conjunto de datos Dog-Pose y sus creadores, visita el sitio web de Stanford Dogs Dataset.
FAQ#
El conjunto de datos Dog-Pose cuenta con 6773 imágenes de entrenamiento y 1703 de validación anotadas con 24 puntos clave para la estimación de postura de perros. Está diseñado para entrenar y validar modelos con Ultralytics YOLO26, permitiendo aplicaciones como el análisis de comportamiento animal, la supervisión de mascotas y estudios veterinarios. Las exhaustivas anotaciones del conjunto de datos lo hacen ideal para desarrollar modelos precisos de estimación de postura para cánidos.
Carga
yolo26n-pose.pty llama amodel.train(data="dog-pose.yaml", epochs=100, imgsz=640)— consulta el Train Example anterior para ver los fragmentos completos de Python y CLI, y la página de Training del modelo para ver una lista exhaustiva de argumentos.Con un total de 8476 imágenes (6773 de entrenamiento / 1703 de validación) que abarcan una amplia gama de razas y posturas de perros, y 24 puntos clave en 3 dimensiones (x, y, visibilidad) por anotación, el conjunto de datos Dog-Pose proporciona a los modelos la cobertura de escenarios del mundo real necesaria para aplicaciones como la supervisión de mascotas y el análisis de comportamiento. Para obtener más información sobre sus características y uso, consulta la sección de Introducción al conjunto de datos.
El mosaico combina varias imágenes de Dog-Pose en una sola imagen de entrenamiento, lo que aumenta la variedad de posturas, tamaños y fondos de perros que el modelo ve en cada paso, mejorando así la generalización a nuevos contextos y escalas a la vez que reduce el sobreajuste. Para ver ejemplos de imágenes, consulta la sección Imágenes de muestra y anotaciones.
El archivo YAML del conjunto de datos Dog-Pose se puede encontrar en https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/dog-pose.yaml. Este archivo define la configuración del conjunto de datos, incluyendo rutas, clases, detalles de puntos clave y otra información relevante. El archivo YAML especifica 24 puntos clave con 3 dimensiones por punto clave, lo que lo hace adecuado para tareas detalladas de estimación de postura.
Para usar este archivo con los scripts de entrenamiento de YOLO26, simplemente haz referencia a él en tu comando de entrenamiento como se muestra en la sección de Uso. El conjunto de datos se descargará automáticamente la primera vez que se use, lo que simplifica la configuración.
Para obtener más información sobre los modelos de puntos clave, consulta la documentación de la tarea de Estimación de poses.



