コンテンツへスキップ

ポーズ推定

ポーズ推定例

姿勢推定は、通常キーポイントと呼ばれる画像内の特定の点の位置を特定するタスクである。キーポイントは、関節、ランドマーク、または他の特徴的な特徴のようなオブジェクトの様々な部分を表すことができます。キーポイントの位置は通常、2次元の [x, y] または3D [x, y, visible] の座標である。

ポーズ推定モデルの出力は、画像内のオブジェクトのキーポイントを表す点の集合であり、通常は各点の信頼度スコアとともに出力される。ポーズ推定は、シーン内のオブジェクトの特定の部分と、それらの位置関係を特定する必要がある場合に適しています。


見るんだ: Ultralytics YOLO によるポーズ推定 .

見るんだ: Ultralytics HUBによる姿勢推定。

チップ

YOLO11 ポーズ モデルは -pose サフィックス、すなわち yolo11n-pose.pt.これらのモデルは COCOのキーポイント データセットであり、様々な姿勢推定タスクに適している。

デフォルトのYOLO11ポーズモデルでは、17個のキーポイントがあり、それぞれが人体の異なる部分を表しています。以下は、各インデックスとそれぞれの身体関節のマッピングです:

0: 鼻 1: 左目 2:右目 3:左耳 4: 右耳 5: 左肩 6: 右肩 7: 左肘 8: 右肘 9: 左手首 10: 右手首 11: 左腰 12: 右腰 13: 左膝 14: 右膝 15: 左足首 16: 右足首

モデル

YOLO11で事前学習されたPoseモデルをここに示す。Detect、Segment、PoseモデルはCOCOデータセットで事前学習され、ClassifyモデルはImageNetデータセットで事前学習されています。

モデルは、初回使用時に最新のUltralytics リリースから自動的にダウンロードされます。

モデルサイズ
(ピクセル)
mAPpose
50-95
mAPpose
50
速度
CPU ONNX
(ms
)
スピード
T4TensorRT10
(ms)
params
(M)
フロップス
(B)
YOLO11n-ポーズ64050.081.052.4 ± 0.51.7 ± 0.02.97.6
YOLO11sポーズ64058.986.390.5 ± 0.62.6 ± 0.09.923.2
YOLO11mポーズ64064.989.4187.3 ± 0.84.9 ± 0.120.971.7
YOLO11l-ポーズ64066.189.9247.7 ± 1.16.4 ± 0.126.290.7
YOLO11xポーズ64069.591.1488.0 ± 13.912.1 ± 0.258.8203.3
  • マップバル 数値はシングル・モデル、シングル・スケールのものである。 COCOキーポイントval2017 データセット。
    複製する yolo val pose data=coco-pose.yaml device=0
  • スピード を使用してCOCOバル画像を平均化した。 アマゾンEC2 P4d インスタンスだ。
    複製する yolo val pose data=coco-pose.yaml batch=1 device=0|cpu

電車

COCO8-poseデータセットでYOLO11-poseモデルをトレーニングする。

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

データセット形式

YOLO ポーズデータセットの形式は、データセットガイドに詳しく書かれている。既存のデータセットを他のフォーマット(COCOなど)からYOLO フォーマットに変換するには、Ultralytics のJSON2YOLOツールをご利用ください。

バル

訓練されたYOLO11n-poseモデルの検証 精度 をCOCO8-poseデータセットに適用した。引数は必要ない。 model トレーニング data と引数をモデル属性として使用する。

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
yolo pose val model=yolo11n-pose.pt  # val official model
yolo pose val model=path/to/best.pt  # val custom model

予測する

訓練されたYOLO11n-poseモデルを使って、画像の予測を実行する。

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
yolo pose predict model=yolo11n-pose.pt source='https://ultralytics.com/images/bus.jpg'  # predict with official model
yolo pose predict model=path/to/best.pt source='https://ultralytics.com/images/bus.jpg'  # predict with custom model

詳細を見る predict モードの詳細は 予測する ページを参照されたい。

輸出

YOLO11nポーズモデルを、ONNX 、CoreML などの異なるフォーマットにエクスポートします。

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")
yolo export model=yolo11n-pose.pt format=onnx  # export official model
yolo export model=path/to/best.pt format=onnx  # export custom trained model

利用可能なYOLO11-ポーズのエクスポート形式は、以下の表の通りです。どのフォーマットでも format 引数、すなわち format='onnx' または format='engine'.エクスポートされたモデルを直接予測または検証することができます。 yolo predict model=yolo11n-pose.onnx.使用例は、エクスポート完了後に表示されます。

フォーマットformat 議論モデルメタデータ議論
PyTorch-yolo11n-pose.pt-
TorchScripttorchscriptyolo11n-pose.torchscriptimgsz, optimize, batch
ONNXonnxyolo11n-pose.onnximgsz, half, dynamic, simplify, opset, batch
OpenVINOopenvinoyolo11n-pose_openvino_model/imgsz, half, int8, batch
TensorRTengineyolo11n-pose.engineimgsz, half, dynamic, simplify, workspace, int8, batch
CoreMLcoremlyolo11n-pose.mlpackageimgsz, half, int8, nms, batch
TF SavedModelsaved_modelyolo11n-pose_saved_model/imgsz, keras, int8, batch
TF GraphDefpbyolo11n-pose.pbimgsz, batch
TF ライトtfliteyolo11n-pose.tfliteimgsz, half, int8, batch
TF エッジTPUedgetpuyolo11n-pose_edgetpu.tfliteimgsz
TF.jstfjsyolo11n-pose_web_model/imgsz, half, int8, batch
PaddlePaddlepaddleyolo11n-pose_paddle_model/imgsz, batch
MNNmnnyolo11n-pose.mnnimgsz, batch, int8, half
NCNNncnnyolo11n-pose_ncnn_model/imgsz, half, batch
IMX500imxyolo11n-pose_imx_model/imgsz, int8

詳細を見る export 詳細は 輸出 ページを参照されたい。

よくあるご質問

Ultralytics YOLO11によるポーズ推定とは何ですか?

Ultralytics YOLO11による姿勢推定では、画像内のキーポイントと呼ばれる特定の点を特定する。これらのキーポイントは通常、関節やオブジェクトの他の重要な特徴を表しています。出力には [x, y] 座標と各ポイントの信頼スコア。YOLO11-ポーズ・モデルは、このタスクのために特別に設計されており -pose のような接尾辞をつける。 yolo11n-pose.pt.これらのモデルは、次のようなデータセットで事前に訓練されています。 COCOのキーポイント 様々なポーズ推定タスクに使用できます。詳しくは ポーズ推定ページ.

カスタムデータセットでYOLO11-poseモデルをトレーニングするには?

YOLO11-poseモデルをカスタムデータセットでトレーニングするには、YAMLファイルで定義された新しいモデルか、事前にトレーニングされたモデルをロードします。その後、指定したデータセットとパラメータを使用してトレーニングプロセスを開始することができます。

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)

トレーニングの包括的な詳細については、「トレーニング」の項を参照のこと。

訓練されたYOLO11ポーズモデルを検証するには?

YOLO11ポーズ・モデルの検証では、トレーニング時に保持された同じデータセット・パラメーターを用いて、その精度を評価する。以下はその例である:

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

詳しくはバル・セクションをご覧ください。

YOLO11のポーズモデルを他のフォーマットにエクスポートできますか?

はい、YOLO11-ポーズモデルを、ONNX 、CoreML 、TensorRT など、さまざまなフォーマットにエクスポートできます。これは、Python またはコマンド・ライン・インターフェース (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")

詳しくはエクスポートの項を参照。

Ultralytics YOLO11ポゼッションモデルとそのパフォーマンス指標は?

Ultralytics YOLO11は、YOLO11n-pose、YOLO11s-pose、YOLO11m-poseなど、様々な事前学習済みポーズモデルを提供しています。これらのモデルは、サイズ、精度(mAP)、速度が異なります。例えば、YOLO11n-poseモデルは、mAPpose50-95で50.4、mAPpose50で80.1を達成している。完全なリストと性能の詳細については、モデルセクションをご覧ください。

📅作成 1年前 ✏️更新:1ヶ月前

コメント