Stima della posa
La stima della posa è un'attività che comporta l'identificazione della posizione di punti specifici in un'immagine, solitamente denominati punti chiave. I punti chiave possono rappresentare varie parti dell'oggetto, come articolazioni, punti di riferimento o altre caratteristiche distintive. Le posizioni dei punti chiave sono di solito rappresentate come un insieme di elementi 2D [x, y]
o 3D [x, y, visible]
coordinate.
L'output di un modello di stima della posa è un insieme di punti che rappresentano i punti chiave di un oggetto nell'immagine, di solito insieme ai punteggi di confidenza per ciascun punto. La stima della posa è una buona scelta quando è necessario identificare parti specifiche di un oggetto in una scena e la loro posizione rispetto agli altri.
Guarda: Ultralytics YOLO11 Esercitazione sulla stima della posa e sul tracciamento degli oggetti in tempo reale e sul rilevamento della posa umana
Suggerimento
YOLO11 posa i modelli utilizzano il -pose
suffisso, cioè yolo11n-pose.pt
. Questi modelli sono addestrati sulla Punti chiave COCO e sono adatti a una varietà di compiti di stima della posa.
Nel modello di posa predefinito di YOLO11 ci sono 17 punti chiave, ognuno dei quali rappresenta una parte diversa del corpo umano. Ecco la mappatura di ciascun indice alla rispettiva articolazione del corpo:
- Naso
- Occhio sinistro
- Occhio destro
- Orecchio sinistro
- Orecchio destro
- Spalla sinistra
- Spalla destra
- Gomito sinistro
- Gomito destro
- Polso sinistro
- Polso destro
- Anca sinistra
- Anca destra
- Ginocchio sinistro
- Ginocchio destro
- Caviglia sinistra
- Caviglia destra
Modelli
I modelli Pose preaddestrati da Ultralytics YOLO11 sono mostrati qui. I modelli Detect, Segment e Pose sono stati preaddestrati sul dataset COCO, mentre i modelli Classify sono stati preaddestrati sul dataset ImageNet.
I modelli vengono scaricati automaticamente dall'ultimarelease di Ultralytics al primo utilizzo.
Modello | dimensione (pixel) |
mAPpose 50-95 |
mAPpose 50 |
Velocità CPU ONNX (ms) |
Velocità T4TensorRT10 (ms) |
params (M) |
FLOP (B) |
---|---|---|---|---|---|---|---|
YOLO11n-pose | 640 | 50.0 | 81.0 | 52.4 ± 0.5 | 1.7 ± 0.0 | 2.9 | 7.6 |
YOLO11s-pose | 640 | 58.9 | 86.3 | 90.5 ± 0.6 | 2.6 ± 0.0 | 9.9 | 23.2 |
YOLO11m-pose | 640 | 64.9 | 89.4 | 187.3 ± 0.8 | 4.9 ± 0.1 | 20.9 | 71.7 |
YOLO11l-pose | 640 | 66.1 | 89.9 | 247.7 ± 1.1 | 6.4 ± 0.1 | 26.2 | 90.7 |
YOLO11x-pose | 640 | 69.5 | 91.1 | 488.0 ± 13.9 | 12.1 ± 0.2 | 58.8 | 203.3 |
- mAPval I valori sono per un modello singolo a scala singola su Punti chiave COCO val2017 set di dati.
Riproduzione dayolo val pose data=coco-pose.yaml device=0
- Velocità mediata sulle immagini COCO val utilizzando una Amazon EC2 P4d istanza.
Riproduzione dayolo val pose data=coco-pose.yaml batch=1 device=0|cpu
Treno
Addestrare un modello YOLO11 sul dataset COCO8-pose. Il set di dati COCO8-pose è un piccolo set di dati campione, perfetto per testare e verificare i modelli di stima della posa.
Esempio
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-pose.yaml") # build a new model from YAML
model = YOLO("yolo11n-pose.pt") # load a pretrained model (recommended for training)
model = YOLO("yolo11n-pose.yaml").load("yolo11n-pose.pt") # build from YAML and transfer weights
# Train the model
results = model.train(data="coco8-pose.yaml", epochs=100, imgsz=640)
# Build a new model from YAML and start training from scratch
yolo pose train data=coco8-pose.yaml model=yolo11n-pose.yaml epochs=100 imgsz=640
# Start training from a pretrained *.pt model
yolo pose train data=coco8-pose.yaml model=yolo11n-pose.pt epochs=100 imgsz=640
# Build a new model from YAML, transfer pretrained weights to it and start training
yolo pose train data=coco8-pose.yaml model=yolo11n-pose.yaml pretrained=yolo11n-pose.pt epochs=100 imgsz=640
Formato del set di dati
Il formato del set di dati della posa YOLO è descritto in dettaglio nella Guida ai set di dati. Per convertire i dataset esistenti da altri formati (come COCO ecc.) al formato YOLO , utilizzare lo strumento JSON2YOLO di Ultralytics.
Per compiti di stima della posa personalizzati, è possibile esplorare set di dati specializzati come Tiger-Pose per la stima della posa degli animali, Hand Keypoints per il tracciamento delle mani o Dog-Pose per l'analisi della posa dei cani.
Val
Convalida del modello YOLO11n-pose addestrato precisione sul set di dati COCO8-pose. Non sono necessari argomenti, poiché il metodo model
mantiene la sua formazione data
e gli argomenti come attributi del modello.
Esempio
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-pose.pt") # load an official model
model = YOLO("path/to/best.pt") # load a custom model
# Validate the model
metrics = model.val() # no arguments needed, dataset and settings remembered
metrics.box.map # map50-95
metrics.box.map50 # map50
metrics.box.map75 # map75
metrics.box.maps # a list contains map50-95 of each category
Prevedere
Utilizzate un modello YOLO11n-pose addestrato per eseguire previsioni sulle immagini. La modalità di predizione consente di eseguire inferenze su immagini, video o flussi in tempo reale.
Esempio
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-pose.pt") # load an official model
model = YOLO("path/to/best.pt") # load a custom model
# Predict with the model
results = model("https://ultralytics.com/images/bus.jpg") # predict on an image
# Access the results
for result in results:
xy = result.keypoints.xy # x and y coordinates
xyn = result.keypoints.xyn # normalized
kpts = result.keypoints.data # x, y, visibility (if available)
Vedi tutto predict
dettagli sulla modalità nella sezione Prevedere pagina.
Esportazione
Esportazione di un modello YOLO11n Pose in un formato diverso come ONNX, CoreML, ecc. Ciò consente di distribuire il modello su varie piattaforme e dispositivi per l'inferenza in tempo reale.
Esempio
I formati di esportazione disponibili di YOLO11-pose sono riportati nella tabella seguente. È possibile esportare in qualsiasi formato utilizzando l'opzione format
cioè l'argomento format='onnx'
o format='engine'
. È possibile prevedere o convalidare direttamente i modelli esportati, ad es. yolo predict model=yolo11n-pose.onnx
. Al termine dell'esportazione vengono mostrati esempi di utilizzo del modello.
Formato | format Argomento |
Modello | Metadati | Argomenti |
---|---|---|---|---|
PyTorch | - | yolo11n-pose.pt |
✅ | - |
TorchScript | torchscript |
yolo11n-pose.torchscript |
✅ | imgsz , optimize , nms , batch |
ONNX | onnx |
yolo11n-pose.onnx |
✅ | imgsz , half , dynamic , simplify , opset , nms , batch |
OpenVINO | openvino |
yolo11n-pose_openvino_model/ |
✅ | imgsz , half , dynamic , int8 , nms , batch , data |
TensorRT | engine |
yolo11n-pose.engine |
✅ | imgsz , half , dynamic , simplify , workspace , int8 , nms , batch , data |
CoreML | coreml |
yolo11n-pose.mlpackage |
✅ | imgsz , half , int8 , nms , batch |
TF SavedModel | saved_model |
yolo11n-pose_saved_model/ |
✅ | imgsz , keras , int8 , nms , batch |
TF GraphDef | pb |
yolo11n-pose.pb |
❌ | imgsz , batch |
TF Lite | tflite |
yolo11n-pose.tflite |
✅ | imgsz , half , int8 , nms , batch , data |
TF Bordo TPU | edgetpu |
yolo11n-pose_edgetpu.tflite |
✅ | imgsz |
TF.js | tfjs |
yolo11n-pose_web_model/ |
✅ | imgsz , half , int8 , nms , batch |
PaddlePaddle | paddle |
yolo11n-pose_paddle_model/ |
✅ | imgsz , batch |
MNN | mnn |
yolo11n-pose.mnn |
✅ | imgsz , batch , int8 , half |
NCNN | ncnn |
yolo11n-pose_ncnn_model/ |
✅ | imgsz , half , batch |
IMX500 | imx |
yolo11n-pose_imx_model/ |
✅ | imgsz , int8 , data |
RKNN | rknn |
yolo11n-pose_rknn_model/ |
✅ | imgsz , batch , name |
Vedi tutto export
dettagli nel Esportazione pagina.
FAQ
Che cos'è la stima della posa con Ultralytics YOLO11 e come funziona?
La stima della posa con Ultralytics YOLO11 comporta l'identificazione di punti specifici, noti come punti chiave, in un'immagine. Questi punti chiave rappresentano in genere articolazioni o altre caratteristiche importanti dell'oggetto. L'output include il valore [x, y]
coordinate e punteggi di confidenza per ogni punto. YOLO11 I modelli -pose sono stati progettati specificamente per questo compito e utilizzano il metodo -pose
come il suffisso yolo11n-pose.pt
. Questi modelli sono pre-addestrati su set di dati come Punti chiave COCO e può essere utilizzato per varie attività di stima della posa. Per ulteriori informazioni, visitate il sito Pagina di stima della posa.
Come posso addestrare un modello YOLO11-pose su un set di dati personalizzato?
L'addestramento di un modello YOLO11-pose su un dataset personalizzato comporta il caricamento di un modello, sia esso un nuovo modello definito da un file YAML o un modello pre-addestrato. È quindi possibile avviare il processo di addestramento utilizzando il set di dati e i parametri specificati.
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-pose.yaml") # build a new model from YAML
model = YOLO("yolo11n-pose.pt") # load a pretrained model (recommended for training)
# Train the model
results = model.train(data="your-dataset.yaml", epochs=100, imgsz=640)
Per maggiori dettagli sull'addestramento, consultare la sezione Addestramento. È inoltre possibile utilizzare Ultralytics HUB per un approccio senza codice all'addestramento di modelli di stima della posa personalizzati.
Come si convalida un modello addestrato YOLO11?
La convalida di un modello YOLO11 comporta la valutazione della sua accuratezza utilizzando gli stessi parametri del set di dati conservati durante l'addestramento. Ecco un esempio:
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-pose.pt") # load an official model
model = YOLO("path/to/best.pt") # load a custom model
# Validate the model
metrics = model.val() # no arguments needed, dataset and settings remembered
Per ulteriori informazioni, visitate la sezione Val.
È possibile esportare un modello YOLO11-pose in altri formati e come?
Sì, è possibile esportare un modello YOLO11 in vari formati, come ONNX, CoreML, TensorRT e altri. È possibile farlo utilizzando Python o l'interfaccia a riga di comando (CLI).
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n-pose.pt") # load an official model
model = YOLO("path/to/best.pt") # load a custom trained model
# Export the model
model.export(format="onnx")
Per maggiori dettagli, consultare la sezione Esportazione. I modelli esportati possono essere distribuiti su dispositivi edge per applicazioni in tempo reale come il fitness tracking, l'analisi sportiva o la robotica.
Quali sono i modelli Ultralytics YOLO11 -pose disponibili e le loro metriche di prestazione?
Ultralytics YOLO11 offre diversi modelli di posa preaddestrati, come YOLO11n-pose, YOLO11s-pose, YOLO11m-pose e altri. Questi modelli differiscono per dimensioni, precisione (mAP) e velocità. Per esempio, il modello YOLO11n-pose raggiunge un mAPpose50-95di 50,0 e un mAPpose50di 81,0. Per un elenco completo e per i dettagli sulle prestazioni, visitare la sezione Modelli.