Estimativa de Pose

A estimativa de pose é uma tarefa que envolve a identificação da localização de pontos específicos em uma imagem, geralmente chamados de pontos-chave (keypoints). Os pontos-chave podem representar várias partes do objeto, como articulações, marcos ou outras características distintas. As localizações dos pontos-chave são geralmente representadas como um conjunto de coordenadas 2D [x, y] ou 3D [x, y, visible].

A saída de um modelo de estimativa de pose é um conjunto de pontos que representam os pontos-chave de um objeto na imagem, geralmente acompanhados pelas pontuações de confiança para cada ponto. A estimativa de pose é uma ótima escolha quando você precisa identificar partes específicas de um objeto em uma cena e a localização delas em relação umas às outras.

Modelos

Modelos de Pose pré-treinados do Ultralytics YOLO26 são exibidos aqui. Os modelos Detect, Segment e Pose são pré-treinados no conjunto de dados COCO, os modelos Semantic são pré-treinados no Cityscapes, e os modelos Classify são pré-treinados no conjunto de dados ImageNet.

Os modelos são baixados automaticamente do lançamento mais recente da Ultralytics na primeira utilização.

• Os valores de mAP^val são para modelo único em escala única no conjunto de dados COCO Keypoints val2017.
  Reproduza com yolo val pose data=coco-pose.yaml device=0
• Velocidade calculada com base na média das imagens de validação do COCO usando uma instância Amazon EC2 P4d.
  Reproduza com yolo val pose data=coco-pose.yaml batch=1 device=0|cpu
• Valores de Params e FLOPs são para o modelo fundido após model.fuse(), que mescla as camadas Conv e BatchNorm e, para modelos end2end, remove a cabeça de detecção auxiliar um-para-muitos. Checkpoints pré-treinados retêm a arquitetura de treinamento completa e podem apresentar contagens mais altas.

Treinar

Treine um modelo YOLO26-pose no conjunto de dados COCO8-pose. O conjunto de dados COCO8-pose é um pequeno conjunto de dados de amostra que é perfeito para testar e depurar seus modelos de estimativa de pose.

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-pose.yaml")  # build a new model from YAML
model = YOLO("yolo26n-pose.pt")  # load a pretrained model (recommended for training)
model = YOLO("yolo26n-pose.yaml").load("yolo26n-pose.pt")  # build from YAML and transfer weights

# Train the model
results = model.train(data="coco8-pose.yaml", epochs=100, imgsz=640)

Veja detalhes completos do modo train na página Train. Modelos de pose também podem ser treinados em GPUs na nuvem através da Ultralytics Platform.

Formato do conjunto de dados

O formato do conjunto de dados de pose do YOLO pode ser encontrado em detalhes no Guia de Conjuntos de Dados. Para converter seu conjunto de dados existente de outros formatos (como COCO, etc.) para o formato YOLO, use a ferramenta JSON2YOLO da Ultralytics. A Ultralytics Platform também oferece suporte à anotação de pose com modelos de esqueleto integrados para pessoas, mãos, rostos e layouts de pontos-chave personalizados.

Para tarefas personalizadas de estimativa de pose, você também pode explorar conjuntos de dados especializados como Tiger-Pose para estimativa de pose animal, Hand Keypoints para rastreamento de mãos ou Dog-Pose para análise de pose canina.

Validação

Valide a precisão do modelo YOLO26n-pose treinado no conjunto de dados COCO8-pose. Nenhum argumento é necessário, pois o model retém seus dados de treinamento data e argumentos como atributos do modelo.

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-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 containing mAP50-95 for each category
metrics.box.image_metrics  # per-image metrics dictionary for box with precision, recall, F1, TP, FP, and FN
metrics.pose.map  # map50-95(P)
metrics.pose.map50  # map50(P)
metrics.pose.map75  # map75(P)
metrics.pose.maps  # a list containing mAP50-95(P) for each category
metrics.pose.image_metrics  # per-image metrics dictionary for pose with precision, recall, F1, TP, FP, and FN

Prever

Use um modelo YOLO26n-pose treinado para realizar previsões em imagens. O predict mode permite que você execute inferência em imagens, vídeos ou transmissões em tempo real.

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-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)

Veja detalhes completos do modo predict na página Predict.

Exportar

Exporte um modelo YOLO26n Pose para um formato diferente, como ONNX, CoreML, etc. Isso permite que você implante seu modelo em várias plataformas e dispositivos para inferência em tempo real.

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-pose.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom-trained model

# Export the model
model.export(format="onnx")

Os formatos de exportação disponíveis para o YOLO26-pose estão na tabela abaixo. Você pode exportar para qualquer formato usando o argumento format, isto é, format='onnx' ou format='engine'. Você pode prever ou validar diretamente em modelos exportados, isto é, yolo predict model=yolo26n-pose.onnx. Exemplos de uso são mostrados para seu modelo após a conclusão da exportação.

Veja detalhes completos de export na página de Exportação.

FAQ

O que é Estimativa de Pose com o Ultralytics YOLO26 e como funciona?

A estimativa de pose com o Ultralytics YOLO26 envolve a identificação de pontos específicos, conhecidos como pontos-chave (keypoints), em uma imagem. Esses pontos-chave normalmente representam articulações ou outras características importantes do objeto. A saída inclui as coordenadas [x, y] e pontuações de confiança para cada ponto. Os modelos YOLO26-pose são projetados especificamente para esta tarefa e usam o sufixo -pose, como yolo26n-pose.pt. Esses modelos são pré-treinados em conjuntos de dados como COCO keypoints e podem ser usados para várias tarefas de estimativa de pose. Para mais informações, visite a Página de Estimativa de Pose.

Como posso treinar um modelo YOLO26-pose em um conjunto de dados personalizado?

Treinar um modelo YOLO26-pose em um conjunto de dados personalizado envolve carregar um modelo, seja um novo modelo definido por um arquivo YAML ou um modelo pré-treinado. Você pode então iniciar o processo de treinamento usando seu conjunto de dados especificado e parâmetros.

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-pose.yaml")  # build a new model from YAML
model = YOLO("yolo26n-pose.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="your-dataset.yaml", epochs=100, imgsz=640)

Para detalhes abrangentes sobre treinamento, consulte a Seção de Treinamento. Você também pode usar a Ultralytics Platform para uma abordagem sem código para treinar modelos de estimativa de pose personalizados.

Como faço para validar um modelo YOLO26-pose treinado?

A validação de um modelo YOLO26-pose envolve avaliar sua precisão usando os mesmos parâmetros do conjunto de dados retidos durante o treinamento. Aqui está um exemplo:

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-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

Para mais informações, visite a Seção de Validação.

Posso exportar um modelo YOLO26-pose para outros formatos e como?

Sim, você pode exportar um modelo YOLO26-pose para vários formatos como ONNX, CoreML, TensorRT e mais. Isso pode ser feito usando Python ou a Interface de Linha de Comando (CLI).

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-pose.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom-trained model

# Export the model
model.export(format="onnx")

Consulte a Seção de Exportação para mais detalhes. Modelos exportados podem ser implantados em dispositivos de borda (edge) para aplicações em tempo real, como rastreamento de fitness, análise esportiva ou robótica.

Quais são os modelos Ultralytics YOLO26-pose disponíveis e suas métricas de desempenho?

O Ultralytics YOLO26 oferece vários modelos de pose pré-treinados, como o YOLO26n-pose, YOLO26s-pose, YOLO26m-pose, entre outros. Estes modelos diferem em tamanho, precisão (mAP) e velocidade. Por exemplo, o modelo YOLO26n-pose alcança um mAP^pose50-95 de 50.0 e um mAP^pose50 de 81.0. Para obter uma lista completa e detalhes de desempenho, visite a Seção de Modelos.