Dog-Pose 데이터셋
소개
Ultralytics Dog-Pose 데이터셋은 개 키포인트 추정을 위해 특별히 선별된 고품질의 방대한 데이터셋입니다. 6,773개의 학습 이미지와 1,703개의 테스트 이미지로 구성된 이 데이터셋은 강력한 자세 추정 모델을 학습시키기 위한 견고한 기반을 제공합니다.
Watch: How to Train Ultralytics YOLO26 on the Stanford Dog Pose Estimation Dataset | Step-by-Step Tutorial
각 어노테이션된 이미지에는 키포인트당 3차원(x, y, 가시성)으로 구성된 24개의 키포인트가 포함되어 있어, 컴퓨터 비전 분야의 고급 연구 및 개발에 귀중한 자원이 됩니다.
이 데이터셋은 Ultralytics Platform 및 YOLO26과 함께 사용하도록 의도되었습니다.
데이터셋 구조
-
분할: 6,773개의 학습 이미지 / 1,703개의 테스트 이미지(YOLO 형식 라벨 파일 포함).
-
Keypoints: 24 per dog with
(x, y, visibility)triplets. -
레이아웃:
datasets/dog-pose/ ├── images/{train,test} └── labels/{train,test}
데이터셋 YAML
데이터셋 구성을 정의하기 위해 YAML(Yet Another Markup Language) 파일이 사용됩니다. 이 파일에는 경로, 키포인트 세부 정보 및 기타 관련 정보가 포함되어 있습니다. Dog-pose 데이터셋의 경우, dog-pose.yaml 파일을 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.zip사용법
Dog-pose 데이터셋으로 YOLO26n-pose 모델을 100 epochs 동안 이미지 크기 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)샘플 이미지 및 주석
다음은 Dog-pose 데이터셋의 이미지 예시와 해당 어노테이션입니다:
- 모자이크 이미지: 이 이미지는 모자이크 처리된 데이터셋 이미지들로 구성된 학습 배치를 보여줍니다. 모자이킹은 학습 중에 여러 이미지를 하나의 이미지로 결합하여 각 학습 배치 내의 객체와 장면의 다양성을 높이는 데 사용되는 기법입니다. 이는 모델이 다양한 객체 크기, 가로세로 비율 및 문맥으로 일반화하는 능력을 향상시키는 데 도움을 줍니다.
이 예시는 Dog-pose 데이터셋에 포함된 이미지의 다양성과 복잡성, 그리고 학습 과정에서 모자이킹(mosaicing)을 사용할 때 얻을 수 있는 이점을 보여줍니다.
인용 및 감사의 글
연구 또는 개발 작업에 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}
}컴퓨터 비전 커뮤니티를 위해 이 귀중한 자원을 만들고 유지 관리해 준 Stanford 팀에 감사를 표합니다. Dog-pose 데이터셋과 제작자에 대한 자세한 정보는 Stanford Dogs Dataset 웹사이트를 방문하십시오.
FAQ
Dog-pose 데이터셋은 무엇이며, Ultralytics YOLO26과 함께 어떻게 사용됩니까?
Dog-Pose 데이터셋은 개 자세 추정을 위한 24개의 키포인트가 어노테이션된 6,773개의 학습 이미지와 1,703개의 테스트 이미지로 구성되어 있습니다. 이 데이터셋은 Ultralytics YOLO26 모델을 학습 및 검증하기 위해 설계되었으며, 동물 행동 분석, 반려동물 모니터링, 수의학 연구와 같은 애플리케이션을 지원합니다. 이 데이터셋의 포괄적인 어노테이션은 정확한 개 자세 추정 모델을 개발하는 데 이상적입니다.
Ultralytics에서 Dog-pose 데이터셋을 사용하여 어떻게 YOLO26 모델을 학습합니까?
Dog-pose 데이터셋에서 YOLO26n-pose 모델을 100 에포크 동안 이미지 크기 640으로 학습하려면 다음 예제를 따르십시오:
from ultralytics import YOLO
# Load a model
model = YOLO("yolo26n-pose.pt")
# Train the model
results = model.train(data="dog-pose.yaml", epochs=100, imgsz=640)학습 인자의 전체 목록은 모델 Training 페이지를 참조하십시오.
Dog-pose 데이터셋을 사용할 때의 이점은 무엇입니까?
Dog-pose 데이터셋은 다음과 같은 여러 이점을 제공합니다:
대규모의 다양한 데이터셋: 8,400개가 넘는 이미지로 구성되어 있어 개 자세, 품종 및 상황을 광범위하게 다루므로 강력한 모델 학습 및 평가가 가능합니다.
상세한 키포인트 어노테이션: 각 이미지에는 키포인트당 3차원(x, y, 가시성)으로 구성된 24개의 키포인트가 포함되어 있어 정확한 자세 탐지 모델 학습을 위한 정밀한 어노테이션을 제공합니다.
실제 시나리오: 다양한 환경의 이미지가 포함되어 있어 반려동물 모니터링 및 행동 분석과 같은 실제 애플리케이션에 대한 모델의 일반화 능력을 향상시킵니다.
전이 학습의 이점: 이 데이터셋은 전이 학습 기술과 잘 호환되어, 사람 자세 데이터셋으로 사전 학습된 모델이 개 특유의 특징에 적응할 수 있도록 합니다.
기능 및 사용법에 대한 자세한 내용은 Dataset Introduction 섹션을 참조하십시오.
모자이킹은 Dog-pose 데이터셋을 사용하는 YOLO26 학습 과정에 어떤 이점을 제공합니까?
Dog-pose 데이터셋의 샘플 이미지에서 볼 수 있듯이 모자이킹은 여러 이미지를 하나의 합성 이미지로 병합하여 학습 배치 내의 객체와 장면 다양성을 풍부하게 합니다. 이 기술은 다음과 같은 이점을 제공합니다:
- 각 배치에서 개의 자세, 크기 및 배경의 다양성 증가
- 다양한 상황과 스케일에서 개를 탐지하는 모델의 능력 향상
- 더 다양한 시각적 패턴에 모델을 노출하여 일반화 성능 강화
- 학습 예제의 새로운 조합을 생성하여 과적합 감소
이 접근 방식은 실제 시나리오에서 더 잘 작동하는 강력한 모델로 이어집니다. 예시 이미지는 샘플 이미지 및 어노테이션 섹션을 참조하십시오.
Dog-pose 데이터셋 YAML 파일은 어디서 찾을 수 있으며 어떻게 사용합니까?
Dog-pose 데이터셋 YAML 파일은 https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/dog-pose.yaml에서 확인할 수 있습니다. 이 파일은 경로, 클래스, 키포인트 세부 정보 및 기타 관련 정보를 포함한 데이터셋 구성을 정의합니다. YAML은 키포인트당 3차원으로 구성된 24개의 키포인트를 지정하므로 상세한 자세 추정 작업에 적합합니다.
YOLO26 학습 스크립트와 함께 이 파일을 사용하려면 사용법 섹션에 표시된 대로 학습 명령에서 해당 파일을 참조하기만 하면 됩니다. 데이터셋은 처음 사용할 때 자동으로 다운로드되므로 설정이 간편합니다.
더 많은 FAQ 및 자세한 문서는 Ultralytics Documentation을 방문하십시오.