Ultralytics YOLO27:

Dataset COCO8-Pose#

Introduzione#

Ultralytics COCO8-Pose è un dataset piccolo ma versatile per la stima della posa, composto dalle prime 8 immagini del set COCO train 2017 (4 per l'addestramento, 4 per la convalida), che utilizza uno schema a 17 punti chiave per la singola classe "person". Questo dataset è ideale per testare e sottoporre a debug i modelli di stima della posa o per sperimentare nuovi approcci al rilevamento dei punti chiave. Con 8 immagini, è abbastanza piccolo da essere facilmente gestibile, ma sufficientemente diversificato da testare le pipeline di addestramento alla ricerca di errori e fungere da controllo di integrità prima dell'addestramento sull'intero dataset COCO-Pose.

Struttura del dataset#

  • Immagini totali: 8 (4 train / 4 val).
  • Classi: 1 (person), con 17 tipi di punti chiave per annotazione.
  • Dimensioni del download: ~1 MB.
  • Struttura di directory consigliata: datasets/coco8-pose/images/{train,val} e datasets/coco8-pose/labels/{train,val}, con i punti chiave in formato YOLO memorizzati come file .txt.

Esplora COCO8-Pose sulla piattaforma Ultralytics per sfogliare ogni immagine con i relativi scheletri dei punti chiave, visualizzare le distribuzioni dei punti chiave e delle classi nella scheda Charts e clonarlo per addestrare il tuo modello nel cloud.

YAML del dataset#

Un file YAML viene utilizzato per definire la configurazione del dataset. Contiene informazioni sui percorsi, sulle classi e su altri dati rilevanti del dataset. Nel caso del dataset COCO8-Pose, il file coco8-pose.yaml è mantenuto all'indirizzo https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/coco8-pose.yaml.

ultralytics/cfg/datasets/coco8-pose.yaml
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# COCO8-pose dataset (first 8 images from COCO train2017) by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/pose/coco8-pose
# Example usage: yolo train data=coco8-pose.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── coco8-pose ← downloads here (1 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: coco8-pose # dataset root dir
train: images/train # train images (relative to 'path') 4 images
val: images/val # val images (relative to 'path') 4 images
test: # test images (optional)

# Keypoints
kpt_shape: [17, 3] # number of keypoints, number of dims (2 for x,y or 3 for x,y,visible)
flip_idx: [0, 2, 1, 4, 3, 6, 5, 8, 7, 10, 9, 12, 11, 14, 13, 16, 15]

# Classes
names:
  0: person

# Keypoint names per class
kpt_names:
  0:
    - nose
    - left_eye
    - right_eye
    - left_ear
    - right_ear
    - left_shoulder
    - right_shoulder
    - left_elbow
    - right_elbow
    - left_wrist
    - right_wrist
    - left_hip
    - right_hip
    - left_knee
    - right_knee
    - left_ankle
    - right_ankle

# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8-pose.zip

Utilizzo#

Per addestrare un modello YOLO26n-pose sul dataset COCO8-Pose per 100 epoche con una dimensione delle immagini di 640, puoi utilizzare i seguenti frammenti di codice. Per un elenco completo degli argomenti disponibili, consulta la pagina di addestramento del modello.

Esempio di addestramento
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="coco8-pose.yaml", epochs=100, imgsz=640)

Immagini ed esempi di annotazioni#

Ecco alcuni esempi di immagini del dataset COCO8-Pose, insieme alle annotazioni corrispondenti:

COCO8-pose keypoint estimation dataset mosaic
  • Immagine a mosaico: questa immagine mostra un batch di addestramento composto da immagini del dataset unite a mosaico. La composizione a mosaico è una tecnica utilizzata durante l'addestramento che combina più immagini in un'unica immagine per aumentare la varietà di oggetti e scene all'interno di ogni batch di addestramento. Questo contribuisce a migliorare la capacità del modello di generalizzare a diverse dimensioni degli oggetti, proporzioni e contesti.

Citazioni e ringraziamenti#

Se utilizzi il dataset COCO nelle tue attività di ricerca o sviluppo, cita il seguente articolo:

Citazione
@misc{lin2015microsoft,
      title={Microsoft COCO: Common Objects in Context},
      author={Tsung-Yi Lin and Michael Maire and Serge Belongie and Lubomir Bourdev and Ross Girshick and James Hays and Pietro Perona and Deva Ramanan and C. Lawrence Zitnick and Piotr Dollár},
      year={2015},
      eprint={1405.0312},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Desideriamo ringraziare il COCO Consortium per aver creato e mantenuto questa preziosa risorsa per la comunità della computer vision. Per ulteriori informazioni sul dataset COCO e sui suoi creatori, visita il sito web del dataset COCO.

FAQ#

  • Il dataset COCO8-Pose è un dataset piccolo e versatile per la stima della posa, che include le prime 8 immagini del set COCO train 2017, con 4 immagini per l'addestramento e 4 per la convalida. È progettato per testare e sottoporre a debug i modelli di stima della posa e per sperimentare nuovi approcci al rilevamento dei punti chiave. Questo dataset è ideale per esperimenti rapidi con Ultralytics YOLO26. Per ulteriori dettagli sulla configurazione del dataset, consulta il file YAML del dataset.

  • Carica yolo26n-pose.pt e richiama model.train(data="coco8-pose.yaml", epochs=100, imgsz=640): consulta il Esempio di addestramento precedente per i frammenti completi in Python e CLI, e la pagina Addestramento del modello per un elenco completo degli argomenti.

  • Con 8 immagini totali (4 train / 4 val), 1 classe, uno schema a 17 punti chiave e un download di circa ~1 MB, COCO8-Pose è abbastanza piccolo da poter essere gestito in pochi secondi, ma sufficientemente diversificato da consentire un controllo di integrità della pipeline di addestramento della posa per individuare errori prima di passare all'intero dataset COCO-Pose. Per ulteriori informazioni sulle sue funzionalità e sul suo utilizzo, consulta la sezione Introduzione al dataset.

  • La creazione di mosaici combina più immagini in una sola, aumentando la varietà di oggetti e scene all'interno di ogni batch di addestramento; ciò aiuta il modello a generalizzare tra diverse dimensioni degli oggetti, proporzioni e contesti. Consulta la sezione Immagini ed esempi di annotazioni per un esempio.

  • Il file YAML del dataset COCO8-Pose è disponibile all'indirizzo https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/coco8-pose.yaml. Questo file definisce la configurazione del dataset, inclusi percorsi, classi e altre informazioni rilevanti. Utilizza questo file con gli script di addestramento di YOLO26, come descritto nella sezione Train Example.

    Per ulteriori informazioni sui modelli di keypoint, consulta la documentazione dell'attività Stima della posa.

Commenti