рд╕рд╛рдордЧреНрд░реА рдкрд░ рдЬрд╛рдПрдВ

COCO-рдкреЛрдЬрд╝ рдбреЗрдЯрд╛рд╕реЗрдЯ

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ COCO (рдХреЙрдорди рдСрдмреНрдЬреЗрдХреНрдЯреНрд╕ рдЗрди рдХреЙрдиреНрдЯреЗрдХреНрд╕реНрдЯ) рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рдПрдХ рд╡рд┐рд╢реЗрд╖ рд╕рдВрд╕реНрдХрд░рдг рд╣реИ, рдЬрд┐рд╕реЗ рдкреЛрдЬрд╝ рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдбрд┐рдЬрд╝рд╛рдЗрди рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИред рдпрд╣ COCO рдХреАрдкреЙрдЗрдВрдЯреНрд╕ 2017 рдЫрд╡рд┐рдпреЛрдВ рдФрд░ рд▓реЗрдмрд▓ рдХрд╛ рд▓рд╛рдн рдЙрдард╛рддрд╛ рд╣реИ рдЬреИрд╕реЗ рдореЙрдбрд▓ рдХреЗ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЛ рд╕рдХреНрд╖рдо рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдПYOLO рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдПред

рдореБрджреНрд░рд╛ рдирдореВрдирд╛ рдЫрд╡рд┐

COCO-рдкреЛрдЬрд╝ рдкреНрд░реАрдЯреНрд░реЗрдиреНрдб рдореЙрдбрд▓

рдХреЛ рдЧрдврд╝рдирд╛ рдЖрдХрд╛рд░ рд╡рд╛рд▓рд╛
(рдкрд┐рдХреНрд╕реЗрд▓)
рдорд╛рдирдЪрд┐рддреНрд░рднрдВрдЧрд┐рдорд╛
50-95
рдорд╛рдирдЪрд┐рддреНрд░рднрдВрдЧрд┐рдорд╛
50
рдЧрддрд┐
CPU ONNX
(рдПрдордПрд╕)
рдЧрддрд┐
рдП100 TensorRT
(рдПрдордПрд╕)
рдкрд░рдо
(рдПрдо)
рдлреНрд▓реЙрдк
(рдмреА)
YOLOv8n-рднрдВрдЧрд┐рдорд╛ 640 50.4 80.1 131.8 1.18 3.3 9.2
YOLOv8s-рднрдВрдЧрд┐рдорд╛ 640 60.0 86.2 233.2 1.42 11.6 30.2
YOLOv8m-рднрдВрдЧрд┐рдорд╛ 640 65.0 88.8 456.3 2.00 26.4 81.0
YOLOv8l-рднрдВрдЧрд┐рдорд╛ 640 67.6 90.0 784.5 2.59 44.4 168.6
YOLOv8x-рднрдВрдЧрд┐рдорд╛ 640 69.2 90.2 1607.1 3.73 69.4 263.2
YOLOv8x-рдкреЛрдЬрд╝-p6 1280 71.6 91.2 4088.7 10.04 99.1 1066.4

рдкреНрд░рдореБрдЦ рд╡рд┐рд╢реЗрд╖рддрд╛рдРрдВ

  • COCO-Pose COCO рдХреАрдкреЙрдЗрдВрдЯреНрд╕ 2017 рдбреЗрдЯрд╛рд╕реЗрдЯ рдкрд░ рдмрдирд╛рддрд╛ рд╣реИ рдЬрд┐рд╕рдореЗрдВ рдкреЛрдЬрд╝ рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдХреАрдкреЙрдЗрдВрдЯ рдХреЗ рд╕рд╛рде рд▓реЗрдмрд▓ рдХреА рдЧрдИ 200K рдЫрд╡рд┐рдпрд╛рдВ рд╣реЛрддреА рд╣реИрдВред
  • рдбреЗрдЯрд╛рд╕реЗрдЯ рдорд╛рдирд╡ рдЖрдВрдХрдбрд╝реЛрдВ рдХреЗ рд▓рд┐рдП 17 рдкреНрд░рдореБрдЦ рдмрд┐рдВрджреБрдУрдВ рдХрд╛ рд╕рдорд░реНрдерди рдХрд░рддрд╛ рд╣реИ, рдЬрд┐рд╕рд╕реЗ рд╡рд┐рд╕реНрддреГрдд рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдХреА рд╕реБрд╡рд┐рдзрд╛ рдорд┐рд▓рддреА рд╣реИред
  • COCO рдХреА рддрд░рд╣, рдпрд╣ рдореБрджреНрд░рд╛, рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдСрдмреНрдЬреЗрдХреНрдЯ рдХреАрдкреЙрдЗрдВрдЯ рд╕рдорд╛рдирддрд╛ (OKS) рд╕рд╣рд┐рдд рдорд╛рдирдХреАрдХреГрдд рдореВрд▓реНрдпрд╛рдВрдХрди рдореАрдЯреНрд░рд┐рдХ рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИ, рдЬрд┐рд╕рд╕реЗ рдпрд╣ рдореЙрдбрд▓ рдкреНрд░рджрд░реНрд╢рди рдХреА рддреБрд▓рдирд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреБрдХреНрдд рд╣реЛ рдЬрд╛рддрд╛ рд╣реИред

рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕рдВрд░рдЪрдирд╛

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЛ рддреАрди рд╕рдмрд╕реЗрдЯ рдореЗрдВ рд╡рд┐рднрд╛рдЬрд┐рдд рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИ:

  1. рдЯреНрд░реЗрди2017: рдЗрд╕ рд╕рдмрд╕реЗрдЯ рдореЗрдВ COCO рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕реЗ 118K рдЫрд╡рд┐рдпреЛрдВ рдХрд╛ рдПрдХ рд╣рд┐рд╕реНрд╕рд╛ рд╢рд╛рдорд┐рд▓ рд╣реИ, рдЬреЛ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП рдПрдиреЛрдЯреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИред
  2. Val2017: рдЗрд╕ рд╕рдмрд╕реЗрдЯ рдореЗрдВ рдореЙрдбрд▓ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЗ рджреМрд░рд╛рди рд╕рддреНрдпрд╛рдкрди рдЙрджреНрджреЗрд╢реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХреА рдЬрд╛рдиреЗ рд╡рд╛рд▓реА рдЫрд╡рд┐рдпреЛрдВ рдХрд╛ рдЪрдпрди рд╣реИред
  3. Test2017: рдЗрд╕ рд╕рдмрд╕реЗрдЯ рдореЗрдВ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓реЛрдВ рдХреЗ рдкрд░реАрдХреНрд╖рдг рдФрд░ рдмреЗрдВрдЪрдорд╛рд░реНрдХрд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХреА рдЬрд╛рдиреЗ рд╡рд╛рд▓реА рдЫрд╡рд┐рдпрд╛рдВ рд╢рд╛рдорд┐рд▓ рд╣реИрдВред рдЗрд╕ рд╕рдмрд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдЧреНрд░рд╛рдЙрдВрдб рдЯреНрд░реБрде рдПрдиреЛрдЯреЗрд╢рди рд╕рд╛рд░реНрд╡рдЬрдирд┐рдХ рд░реВрдк рд╕реЗ рдЙрдкрд▓рдмреНрдз рдирд╣реАрдВ рд╣реИрдВ, рдФрд░ рдкрд░рд┐рдгрд╛рдо рдкреНрд░рджрд░реНрд╢рди рдореВрд▓реНрдпрд╛рдВрдХрди рдХреЗ рд▓рд┐рдП COCO рдореВрд▓реНрдпрд╛рдВрдХрди рд╕рд░реНрд╡рд░ рдкрд░ рд╕рдмрдорд┐рдЯ рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред

рдЕрдиреБрдкреНрд░рдпреЛрдЧреЛрдВ

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рдЙрдкрдпреЛрдЧ рд╡рд┐рд╢реЗрд╖ рд░реВрдк рд╕реЗ рдХреАрдкреЙрдЗрдВрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдФрд░ рдкреЛрдЬрд╝ рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдореЗрдВ рдЧрд╣рди рд╢рд┐рдХреНрд╖рдг рдореЙрдбрд▓ рдХреЗ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдФрд░ рдореВрд▓реНрдпрд╛рдВрдХрди рдХреЗ рд▓рд┐рдП рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ, рдЬреИрд╕реЗ OpenPoseред рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреА рдмрдбрд╝реА рд╕рдВрдЦреНрдпрд╛ рдореЗрдВ рдПрдиреЛрдЯреЗрдЯ рдХреА рдЧрдИ рдЫрд╡рд┐рдпрд╛рдВ рдФрд░ рдорд╛рдирдХреАрдХреГрдд рдореВрд▓реНрдпрд╛рдВрдХрди рдореЗрдЯреНрд░рд┐рдХреНрд╕ рдЗрд╕реЗ рдХрдВрдкреНрдпреВрдЯрд░ рджреГрд╖реНрдЯрд┐ рд╢реЛрдзрдХрд░реНрддрд╛рдУрдВ рдФрд░ рдЪрд┐рдХрд┐рддреНрд╕рдХреЛрдВ рдХреЗ рд▓рд┐рдП рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдкрд░ рдХреЗрдВрджреНрд░рд┐рдд рдПрдХ рдЖрд╡рд╢реНрдпрдХ рд╕рдВрд╕рд╛рдзрди рдмрдирд╛рддреЗ рд╣реИрдВред

рдбреЗрдЯрд╛рд╕реЗрдЯ YAML

рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдХреЛ рдкрд░рд┐рднрд╛рд╖рд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ YAML (рдЕрднреА рддрдХ рдПрдХ рдФрд░ рдорд╛рд░реНрдХрдЕрдк рднрд╛рд╖рд╛) рдлрд╝рд╛рдЗрд▓ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИред рдЗрд╕рдореЗрдВ рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рдкрде, рдХрдХреНрд╖рд╛рдУрдВ рдФрд░ рдЕрдиреНрдп рдкреНрд░рд╛рд╕рдВрдЧрд┐рдХ рдЬрд╛рдирдХрд╛рд░реА рдХреЗ рдмрд╛рд░реЗ рдореЗрдВ рдЬрд╛рдирдХрд╛рд░реА рд╣реЛрддреА рд╣реИред COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рдорд╛рдорд▓реЗ рдореЗрдВ, coco-pose.yaml рдлрд╝рд╛рдЗрд▓ рдкрд░ рдмрдирд╛рдП рд░рдЦрд╛ рдЬрд╛рддрд╛ рд╣реИ https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/coco-pose.yaml.

ultralytics/cfg/datasets/coco-pose.yaml

# Ultralytics YOLO ЁЯЪА, AGPL-3.0 license
# COCO 2017 dataset https://cocodataset.org by Microsoft
# Documentation: https://docs.ultralytics.com/datasets/pose/coco/
# Example usage: yolo train data=coco-pose.yaml
# parent
# тФЬтФАтФА ultralytics
# тФФтФАтФА datasets
#     тФФтФАтФА coco-pose  тЖР downloads here (20.1 GB)

# 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: ../datasets/coco-pose # dataset root dir
train: train2017.txt # train images (relative to 'path') 118287 images
val: val2017.txt # val images (relative to 'path') 5000 images
test: test-dev2017.txt # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794

# 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

# Download script/URL (optional)
download: |
  from ultralytics.utils.downloads import download
  from pathlib import Path

  # Download labels
  dir = Path(yaml['path'])  # dataset root dir
  url = 'https://github.com/ultralytics/assets/releases/download/v0.0.0/'
  urls = [url + 'coco2017labels-pose.zip']  # labels
  download(urls, dir=dir.parent)
  # Download data
  urls = ['http://images.cocodataset.org/zips/train2017.zip',  # 19G, 118k images
          'http://images.cocodataset.org/zips/val2017.zip',  # 1G, 5k images
          'http://images.cocodataset.org/zips/test2017.zip']  # 7G, 41k images (optional)
  download(urls, dir=dir / 'images', threads=3)

рдЙрдкрдпреЛрдЧ

рдПрдХ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП YOLOv8n-POSE рдореЙрдбрд▓ 640 рдХреЗ рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдХреЗ рд╕рд╛рде 100 рдпреБрдЧреЛрдВ рдХреЗ рд▓рд┐рдП COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдкрд░, рдЖрдк рдирд┐рдореНрди рдХреЛрдб рд╕реНрдирд┐рдкреЗрдЯ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░ рд╕рдХрддреЗ рд╣реИрдВред рдЙрдкрд▓рдмреНрдз рддрд░реНрдХреЛрдВ рдХреА рд╡реНрдпрд╛рдкрдХ рд╕реВрдЪреА рдХреЗ рд▓рд┐рдП, рдореЙрдбрд▓ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдкреГрд╖реНрда рджреЗрдЦреЗрдВред

рдЯреНрд░реЗрди рдХрд╛ рдЙрджрд╛рд╣рд░рдг

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n-pose.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="coco-pose.yaml", epochs=100, imgsz=640)
# Start training from a pretrained *.pt model
yolo detect train data=coco-pose.yaml model=yolov8n-pose.pt epochs=100 imgsz=640

рдирдореВрдирд╛ рдЫрд╡рд┐рдпрд╛рдБ рдФрд░ рдПрдиреЛрдЯреЗрд╢рди

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдореЗрдВ рдореБрдЦреНрдп рдмрд┐рдВрджреБрдУрдВ рдХреЗ рд╕рд╛рде рдПрдиреЛрдЯреЗрдЯ рдХрд┐рдП рдЧрдП рдорд╛рдирд╡ рдЖрдХреГрддрд┐рдпреЛрдВ рдХреЗ рд╕рд╛рде рдЫрд╡рд┐рдпреЛрдВ рдХрд╛ рдПрдХ рд╡рд┐рд╡рд┐рдз рд╕реЗрдЯ рд╣реЛрддрд╛ рд╣реИред рдпрд╣рд╛рдВ рдбреЗрдЯрд╛рд╕реЗрдЯ рд╕реЗ рдЫрд╡рд┐рдпреЛрдВ рдХреЗ рдХреБрдЫ рдЙрджрд╛рд╣рд░рдг рджрд┐рдП рдЧрдП рд╣реИрдВ, рд╕рд╛рде рд╣реА рдЙрдирдХреЗ рд╕рдВрдмрдВрдзрд┐рдд рдПрдиреЛрдЯреЗрд╢рди рдХреЗ рд╕рд╛рде:

рдбреЗрдЯрд╛рд╕реЗрдЯ рдирдореВрдирд╛ рдЫрд╡рд┐

  • рдореЛрдЬрд╝реЗрдб рдЫрд╡рд┐: рдпрд╣ рдЫрд╡рд┐ рдореЛрдЬрд╝реЗрдХ рдбреЗрдЯрд╛рд╕реЗрдЯ рдЫрд╡рд┐рдпреЛрдВ рд╕реЗ рдмрдирд╛ рдПрдХ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдмреИрдЪ рдкреНрд░рджрд░реНрд╢рд┐рдд рдХрд░рддреА рд╣реИред рдореЛрдЬрд╝реЗрд╕рд┐рдВрдЧ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЗ рджреМрд░рд╛рди рдЙрдкрдпреЛрдЧ рдХреА рдЬрд╛рдиреЗ рд╡рд╛рд▓реА рдПрдХ рддрдХрдиреАрдХ рд╣реИ рдЬреЛ рдкреНрд░рддреНрдпреЗрдХ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдмреИрдЪ рдХреЗ рднреАрддрд░ рд╡рд╕реНрддреБрдУрдВ рдФрд░ рджреГрд╢реНрдпреЛрдВ рдХреА рд╡рд┐рд╡рд┐рдзрддрд╛ рдХреЛ рдмрдврд╝рд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдХрдИ рдЫрд╡рд┐рдпреЛрдВ рдХреЛ рдПрдХ рд╣реА рдЫрд╡рд┐ рдореЗрдВ рдЬреЛрдбрд╝рддреА рд╣реИред рдпрд╣ рд╡рд┐рднрд┐рдиреНрди рдСрдмреНрдЬреЗрдХреНрдЯ рдЖрдХрд╛рд░реЛрдВ, рдкрд╣рд▓реВ рдЕрдиреБрдкрд╛рдд рдФрд░ рд╕рдВрджрд░реНрднреЛрдВ рдХреЛ рд╕рд╛рдорд╛рдиреНрдпреАрдХреГрдд рдХрд░рдиреЗ рдХреА рдореЙрдбрд▓ рдХреА рдХреНрд╖рдорддрд╛ рдХреЛ рдмреЗрд╣рддрд░ рдмрдирд╛рдиреЗ рдореЗрдВ рдорджрдж рдХрд░рддрд╛ рд╣реИред

рдЙрджрд╛рд╣рд░рдг COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдореЗрдВ рдЫрд╡рд┐рдпреЛрдВ рдХреА рд╡рд┐рд╡рд┐рдзрддрд╛ рдФрд░ рдЬрдЯрд┐рд▓рддрд╛ рдФрд░ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдкреНрд░рдХреНрд░рд┐рдпрд╛ рдХреЗ рджреМрд░рд╛рди рдореЛрдЬрд╝реЗрдХрд┐рдВрдЧ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдиреЗ рдХреЗ рд▓рд╛рднреЛрдВ рдХреЛ рдкреНрд░рджрд░реНрд╢рд┐рдд рдХрд░рддрд╛ рд╣реИред

рдкреНрд░рд╢рдВрд╕рд╛ рдкрддреНрд░ рдФрд░ рдкрд╛рд╡рддреА

рдпрджрд┐ рдЖрдк рдЕрдкрдиреЗ рд╢реЛрдз рдпрд╛ рд╡рд┐рдХрд╛рд╕ рдХрд╛рд░реНрдп рдореЗрдВ COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддреЗ рд╣реИрдВ, рддреЛ рдХреГрдкрдпрд╛ рдирд┐рдореНрдирд▓рд┐рдЦрд┐рдд рдкреЗрдкрд░ рдХрд╛ рд╣рд╡рд╛рд▓рд╛ рджреЗрдВ:

@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}
}

рд╣рдо рдХрдВрдкреНрдпреВрдЯрд░ рджреГрд╖реНрдЯрд┐ рд╕рдореБрджрд╛рдп рдХреЗ рд▓рд┐рдП рдЗрд╕ рдореВрд▓реНрдпрд╡рд╛рди рд╕рдВрд╕рд╛рдзрди рдХреЛ рдмрдирд╛рдиреЗ рдФрд░ рдмрдирд╛рдП рд░рдЦрдиреЗ рдХреЗ рд▓рд┐рдП рдХреЛрдХреЛ рдХрдВрд╕реЛрд░реНрдЯрд┐рдпрдо рдХреЛ рд╕реНрд╡реАрдХрд╛рд░ рдХрд░рдирд╛ рдЪрд╛рд╣рддреЗ рд╣реИрдВред COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдФрд░ рдЗрд╕рдХреЗ рд░рдЪрдирд╛рдХрд╛рд░реЛрдВ рдХреЗ рдмрд╛рд░реЗ рдореЗрдВ рдЕрдзрд┐рдХ рдЬрд╛рдирдХрд╛рд░реА рдХреЗ рд▓рд┐рдП, COCO рдбреЗрдЯрд╛рд╕реЗрдЯ рд╡реЗрдмрд╕рд╛рдЗрдЯ рдкрд░ рдЬрд╛рдПрдБред

рдЕрдХреНрд╕рд░ рдкреВрдЫреЗ рдЬрд╛рдиреЗ рд╡рд╛рд▓реЗ рдкреНрд░рд╢реНрди

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреНрдпрд╛ рд╣реИ рдФрд░ рдЗрд╕рдХрд╛ рдЙрдкрдпреЛрдЧ рдХреИрд╕реЗ рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ Ultralytics YOLO рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдП?

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ COCO (рдХреЙрдорди рдСрдмреНрдЬреЗрдХреНрдЯреНрд╕ рдЗрди рдХреЙрдиреНрдЯреЗрдХреНрд╕реНрдЯ) рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рдПрдХ рд╡рд┐рд╢реЗрд╖ рд╕рдВрд╕реНрдХрд░рдг рд╣реИ рдЬрд┐рд╕реЗ рдкреЛрдЬрд╝ рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдбрд┐рдЬрд╝рд╛рдЗрди рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИред рдпрд╣ COCO рдХреАрдкреЙрдЗрдВрдЯреНрд╕ 2017 рдЫрд╡рд┐рдпреЛрдВ рдФрд░ рдПрдиреЛрдЯреЗрд╢рди рдкрд░ рдмрдирд╛рддрд╛ рд╣реИ, рдЬрд┐рд╕рд╕реЗ рдореЙрдбрд▓ рдХреЗ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреА рдЕрдиреБрдорддрд┐ рдорд┐рд▓рддреА рд╣реИUltralytics YOLO рд╡рд┐рд╕реНрддреГрдд рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдПред рдЙрджрд╛рд╣рд░рдг рдХреЗ рд▓рд┐рдП, рдЖрдк рдПрдХ рдХреЛ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░ рд╕рдХрддреЗ рд╣реИрдВ YOLOv8n-рдкреЛрдЬрд╝ рдореЙрдбрд▓ рдХреЛ рдПрдХ рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓ рд▓реЛрдб рдХрд░рдХреЗ рдФрд░ рдЗрд╕реЗ YAML рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рдХреЗ рд╕рд╛рде рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрд░рдХреЗред рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдЙрджрд╛рд╣рд░рдгреЛрдВ рдХреЗ рд▓рд┐рдП, рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдкреНрд░рд▓реЗрдЦрди рджреЗрдЦреЗрдВред

рдореИрдВ рдХреИрд╕реЗ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрд░ рд╕рдХрддрд╛ рд╣реВрдВ YOLOv8 COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдкрд░ рдореЙрдбрд▓?

рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдПрдХ YOLOv8 COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдкрд░ рдореЙрдбрд▓ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдкреВрд░рд╛ рдХрд┐рдпрд╛ рдЬрд╛ рд╕рдХрддрд╛ рд╣реИ Python рдирд╣реАрдВ рддреЛ CLI рдЖрджреЗрд╢ред рдЙрджрд╛рд╣рд░рдг рдХреЗ рд▓рд┐рдП, рдПрдХ рдХреЛ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП YOLOv8n640 рдХреА рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдХреЗ рд╕рд╛рде 100 рдпреБрдЧреЛрдВ рдХреЗ рд▓рд┐рдП -рдкреЛрдЬрд╝ рдореЙрдбрд▓, рдЖрдк рдиреАрдЪреЗ рджрд┐рдП рдЧрдП рдЪрд░рдгреЛрдВ рдХрд╛ рдкрд╛рд▓рди рдХрд░ рд╕рдХрддреЗ рд╣реИрдВ:

рдЯреНрд░реЗрди рдХрд╛ рдЙрджрд╛рд╣рд░рдг

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n-pose.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="coco-pose.yaml", epochs=100, imgsz=640)
# Start training from a pretrained *.pt model
yolo detect train data=coco-pose.yaml model=yolov8n.pt epochs=100 imgsz=640

рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдкреНрд░рдХреНрд░рд┐рдпрд╛ рдФрд░ рдЙрдкрд▓рдмреНрдз рддрд░реНрдХреЛрдВ рдХреЗ рдмрд╛рд░реЗ рдореЗрдВ рдЕрдзрд┐рдХ рдЬрд╛рдирдХрд╛рд░реА рдХреЗ рд▓рд┐рдП, рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдкреГрд╖реНрда рджреЗрдЦреЗрдВред

рдореЙрдбрд▓ рдкреНрд░рджрд░реНрд╢рди рдХреЗ рдореВрд▓реНрдпрд╛рдВрдХрди рдХреЗ рд▓рд┐рдП COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рджреНрд╡рд╛рд░рд╛ рдкреНрд░рджрд╛рди рдХрд┐рдП рдЧрдП рд╡рд┐рднрд┐рдиреНрди рдореАрдЯреНрд░рд┐рдХ рдХреНрдпрд╛ рд╣реИрдВ?

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдореВрд▓ COCO рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд╕рдорд╛рди рдкреЛрдЬрд╝ рдЕрдиреБрдорд╛рди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдХрдИ рдорд╛рдирдХреАрдХреГрдд рдореВрд▓реНрдпрд╛рдВрдХрди рдореАрдЯреНрд░рд┐рдХ рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред рдореБрдЦреНрдп рдореАрдЯреНрд░рд┐рдХ рдореЗрдВ рдСрдмреНрдЬреЗрдХреНрдЯ рдХреАрдкреЙрдЗрдВрдЯ рд╕рдорд╛рдирддрд╛ (рдУрдХреЗрдПрд╕) рд╢рд╛рдорд┐рд▓ рд╣реИ, рдЬреЛ рдЬрдореАрдиреА рд╕рдЪреНрдЪрд╛рдИ рдПрдиреЛрдЯреЗрд╢рди рдХреЗ рдЦрд┐рд▓рд╛рдл рдЕрдиреБрдорд╛рдирд┐рдд рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рдХреА рд╕рдЯреАрдХрддрд╛ рдХрд╛ рдореВрд▓реНрдпрд╛рдВрдХрди рдХрд░рддрд╛ рд╣реИред рдпреЗ рдореАрдЯреНрд░рд┐рдХ рд╡рд┐рднрд┐рдиреНрди рдореЙрдбрд▓реЛрдВ рдХреЗ рдмреАрдЪ рдкреВрд░реА рддрд░рд╣ рд╕реЗ рдкреНрд░рджрд░реНрд╢рди рддреБрд▓рдирд╛ рдХрд░рдиреЗ рдХреА рдЕрдиреБрдорддрд┐ рджреЗрддреЗ рд╣реИрдВред рдЙрджрд╛рд╣рд░рдг рдХреЗ рд▓рд┐рдП, COCO-Pose рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓ рдЬреИрд╕реЗ YOLOv8n-рднрдВрдЧрд┐рдорд╛ YOLOv8s-рдкреЛрдЬрд╝, рдФрд░ рдЕрдиреНрдп рдХреЗ рдкрд╛рд╕ рджрд╕реНрддрд╛рд╡реЗрдЬрд╝реАрдХрд░рдг рдореЗрдВ рд╕реВрдЪреАрдмрджреНрдз рд╡рд┐рд╢рд┐рд╖реНрдЯ рдкреНрд░рджрд░реНрд╢рди рдореАрдЯреНрд░рд┐рдХ рд╣реИрдВ, рдЬреИрд╕реЗ рдПрдордПрдкреАрдкреЛрдЬрд╝50-95 рдФрд░ рдПрдордПрдкреАрдкреЛрдЬрд╝50ред

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЛ рдХреИрд╕реЗ рд╕рдВрд░рдЪрд┐рдд рдФрд░ рд╡рд┐рднрд╛рдЬрд┐рдд рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИ?

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЛ рддреАрди рд╕рдмрд╕реЗрдЯ рдореЗрдВ рд╡рд┐рднрд╛рдЬрд┐рдд рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИ:

  1. рдЯреНрд░реЗрди2017: рдЗрд╕рдореЗрдВ 118K COCO рдЫрд╡рд┐рдпреЛрдВ рдХрд╛ рдПрдХ рд╣рд┐рд╕реНрд╕рд╛ рд╢рд╛рдорд┐рд▓ рд╣реИ, рдЬрд┐рд╕реЗ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдореБрджреНрд░рд╛ рдЕрдиреБрдорд╛рди рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП рдПрдиреЛрдЯреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИред
  2. Val2017: рдореЙрдбрд▓ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЗ рджреМрд░рд╛рди рд╕рддреНрдпрд╛рдкрди рдЙрджреНрджреЗрд╢реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдЪрдпрдирд┐рдд рдЫрд╡рд┐рдпрд╛рдВред
  3. Test2017: рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓реЛрдВ рдХреЗ рдкрд░реАрдХреНрд╖рдг рдФрд░ рдмреЗрдВрдЪрдорд╛рд░реНрдХрд┐рдВрдЧ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреЛрдЧ рдХреА рдЬрд╛рдиреЗ рд╡рд╛рд▓реА рдЫрд╡рд┐рдпрд╛рдВред рдЗрд╕ рд╕рдмрд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдЧреНрд░рд╛рдЙрдВрдб рдЯреНрд░реБрде рдПрдиреЛрдЯреЗрд╢рди рд╕рд╛рд░реНрд╡рдЬрдирд┐рдХ рд░реВрдк рд╕реЗ рдЙрдкрд▓рдмреНрдз рдирд╣реАрдВ рд╣реИрдВ; рдкрд░рд┐рдгрд╛рдо рдкреНрд░рджрд░реНрд╢рди рдореВрд▓реНрдпрд╛рдВрдХрди рдХреЗ рд▓рд┐рдП COCO рдореВрд▓реНрдпрд╛рдВрдХрди рд╕рд░реНрд╡рд░ рдХреЛ рдкреНрд░рд╕реНрддреБрдд рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВред

рдпреЗ рд╕рдмрд╕реЗрдЯ рдкреНрд░рд╢рд┐рдХреНрд╖рдг, рд╕рддреНрдпрд╛рдкрди рдФрд░ рдкрд░реАрдХреНрд╖рдг рдЪрд░рдгреЛрдВ рдХреЛ рдкреНрд░рднрд╛рд╡реА рдврдВрдЧ рд╕реЗ рд╡реНрдпрд╡рд╕реНрдерд┐рдд рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░рддреЗ рд╣реИрдВред рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди рд╡рд┐рд╡рд░рдг рдХреЗ рд▓рд┐рдП, рдПрдХреНрд╕рдкреНрд▓реЛрд░ рдХрд░реЗрдВ coco-pose.yaml рдлрд╝рд╛рдЗрд▓ рдкрд░ рдЙрдкрд▓рдмреНрдз рд╣реИ рдЧрд┐рдЯрд╣рдм.

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреА рдкреНрд░рдореБрдЦ рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдВ рдФрд░ рдЕрдиреБрдкреНрд░рдпреЛрдЧ рдХреНрдпрд╛ рд╣реИрдВ?

COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ COCO рдХреАрдкреЙрдЗрдВрдЯреНрд╕ 2017 рдПрдиреЛрдЯреЗрд╢рди рдХреЛ рдорд╛рдирд╡ рдЖрдВрдХрдбрд╝реЛрдВ рдХреЗ рд▓рд┐рдП 17 рдХреАрдкреЙрдЗрдВрдЯреНрд╕ рд╢рд╛рдорд┐рд▓ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рд╡рд┐рд╕реНрддрд╛рд░рд┐рдд рдХрд░рддрд╛ рд╣реИ, рдЬрд┐рд╕рд╕реЗ рд╡рд┐рд╕реНрддреГрдд рдкреЛрдЬрд╝ рдЕрдиреБрдорд╛рди рд╕рдХреНрд╖рдо рд╣реЛрддрд╛ рд╣реИред рдорд╛рдирдХреАрдХреГрдд рдореВрд▓реНрдпрд╛рдВрдХрди рдореАрдЯреНрд░рд┐рдХ (рдЬреИрд╕реЗ, рдУрдХреЗрдПрд╕) рд╡рд┐рднрд┐рдиреНрди рдореЙрдбрд▓реЛрдВ рдореЗрдВ рддреБрд▓рдирд╛ рдХреА рд╕реБрд╡рд┐рдзрд╛ рдкреНрд░рджрд╛рди рдХрд░рддреЗ рд╣реИрдВред COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рдЕрдиреБрдкреНрд░рдпреЛрдЧ рд╡рд┐рднрд┐рдиреНрди рдбреЛрдореЗрди рдореЗрдВ рдлреИрд▓реЗ рд╣реБрдП рд╣реИрдВ, рдЬреИрд╕реЗ рдХрд┐ рд╕реНрдкреЛрд░реНрдЯреНрд╕ рдПрдирд╛рд▓рд┐рдЯрд┐рдХреНрд╕, рд╣реЗрд▓реНрдердХреЗрдпрд░ рдФрд░ рдорд╛рдирд╡-рдХрдВрдкреНрдпреВрдЯрд░ рдЗрдВрдЯрд░реИрдХреНрд╢рди, рдЬрд╣рд╛рдВ рдорд╛рдирд╡ рдЖрдХреГрддрд┐рдпреЛрдВ рдХреЗ рд╡рд┐рд╕реНрддреГрдд рдЕрдиреБрдорд╛рди рдХреА рдЖрд╡рд╢реНрдпрдХрддрд╛ рд╣реЛрддреА рд╣реИред рд╡реНрдпрд╛рд╡рд╣рд╛рд░рд┐рдХ рдЙрдкрдпреЛрдЧ рдХреЗ рд▓рд┐рдП, рдкреНрд░рд▓реЗрдЦрди рдореЗрдВ рдкреНрд░рджрд╛рди рдХрд┐рдП рдЧрдП рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓ рдХрд╛ рд▓рд╛рдн рдЙрдард╛рдирд╛ (рдЙрджрд╛ред YOLOv8n-рдореБрджреНрд░рд╛) рдкреНрд░рдХреНрд░рд┐рдпрд╛ рдХреЛ рдорд╣рддреНрд╡рдкреВрд░реНрдг рд░реВрдк рд╕реЗ рд╕реБрд╡реНрдпрд╡рд╕реНрдерд┐рдд рдХрд░ рд╕рдХрддрд╛ рд╣реИ (рдкреНрд░рдореБрдЦ рд╡рд┐рд╢реЗрд╖рддрд╛рдПрдВ)ред

рдпрджрд┐ рдЖрдк рдЕрдкрдиреЗ рд╢реЛрдз рдпрд╛ рд╡рд┐рдХрд╛рд╕ рдХрд╛рд░реНрдп рдореЗрдВ COCO-Pose рдбреЗрдЯрд╛рд╕реЗрдЯ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддреЗ рд╣реИрдВ, рддреЛ рдХреГрдкрдпрд╛ рдирд┐рдореНрдирд▓рд┐рдЦрд┐рдд BibTeX рдкреНрд░рд╡рд┐рд╖реНрдЯрд┐ рдХреЗ рд╕рд╛рде рдкреЗрдкрд░ рдХрд╛ рд╣рд╡рд╛рд▓рд╛ рджреЗрдВред



рдмрдирд╛рдпрд╛ рдЧрдпрд╛ 2023-11-12, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-07-17
рд▓реЗрдЦрдХ: hnliu_2@stu.xidian.edu.cn (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (7), рд░рд┐рдЬрд╡рд╛рди рдореБрдирд╡реНрд╡рд░ (1), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1)

рдЯрд┐рдкреНрдкрдгрд┐рдпрд╛рдБ