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

Meituan YOLOv6

рд╡рд┐рд╣рдВрдЧрд╛рд╡рд▓реЛрдХрди

рдореАрддреБрдЖрди YOLOv6 рдПрдХ рдЕрддреНрдпрд╛рдзреБрдирд┐рдХ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрдЯрд░ рд╣реИ рдЬреЛ рдЧрддрд┐ рдФрд░ рд╕рдЯреАрдХрддрд╛ рдХреЗ рдмреАрдЪ рдЙрд▓реНрд▓реЗрдЦрдиреАрдп рд╕рдВрддреБрд▓рди рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИ, рдЬрд┐рд╕рд╕реЗ рдпрд╣ рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдЕрдиреБрдкреНрд░рдпреЛрдЧреЛрдВ рдХреЗ рд▓рд┐рдП рдПрдХ рд▓реЛрдХрдкреНрд░рд┐рдп рд╡рд┐рдХрд▓реНрдк рдмрди рдЬрд╛рддрд╛ рд╣реИред рдпрд╣ рдореЙрдбрд▓ рдЕрдкрдиреА рд╡рд╛рд╕реНрддреБрдХрд▓рд╛ рдФрд░ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдпреЛрдЬрдирд╛ рдкрд░ рдХрдИ рдЙрд▓реНрд▓реЗрдЦрдиреАрдп рд╕рдВрд╡рд░реНрджреНрдзрди рдХрд╛ рдкрд░рд┐рдЪрдп рджреЗрддрд╛ рд╣реИ, рдЬрд┐рд╕рдореЗрдВ рджреНрд╡рд┐-рджрд┐рд╢рд╛рддреНрдордХ рд╕рдВрдпреЛрдЬрди (рдмреАрдЖрдИрд╕реА) рдореЙрдбреНрдпреВрд▓, рдПрдХ рдПрдВрдХрд░-рдПрдбреЗрдб рдкреНрд░рд╢рд┐рдХреНрд╖рдг (рдПрдПрдЯреА) рд░рдгрдиреАрддрд┐, рдФрд░ рдЕрддреНрдпрд╛рдзреБрдирд┐рдХ рд╕рдЯреАрдХрддрд╛ рдХреЗ рд▓рд┐рдП рдПрдХ рдмреЗрд╣рддрд░ рд░реАрдврд╝ рдФрд░ рдЧрд░реНрджрди рдбрд┐рдЬрд╛рдЗрди рд╢рд╛рдорд┐рд▓ рд╣реИред

Meituan YOLOv6 рдореЙрдбрд▓ рдЙрджрд╛рд╣рд░рдг рдЫрд╡рд┐ YOLOv6 рдХрд╛ рдЕрд╡рд▓реЛрдХрдиред рдореЙрдбрд▓ рдЖрд░реНрдХрд┐рдЯреЗрдХреНрдЪрд░ рдЖрд░реЗрдЦ рдкреБрди: рдбрд┐рдЬрд╝рд╛рдЗрди рдХрд┐рдП рдЧрдП рдиреЗрдЯрд╡рд░реНрдХ рдШрдЯрдХреЛрдВ рдФрд░ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рд░рдгрдиреАрддрд┐рдпреЛрдВ рдХреЛ рджрд┐рдЦрд╛ рд░рд╣рд╛ рд╣реИ рдЬрд┐рд╕рд╕реЗ рдорд╣рддреНрд╡рдкреВрд░реНрдг рдкреНрд░рджрд░реНрд╢рди рд╕реБрдзрд╛рд░ рд╣реБрдП рд╣реИрдВред (a) YOLOv6 рдХреА рдЧрд░реНрджрди (N рдФрд░ S рджрд┐рдЦрд╛рдП рдЧрдП рд╣реИрдВ)ред M/L рдХреЗ рд▓рд┐рдП рдиреЛрдЯ, RepBlocks рдХреЛ CSPStackRep рд╕реЗ рдмрджрд▓ рджрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИред (b) BiC рдореЙрдбреНрдпреВрд▓ рдХреА рд╕рдВрд░рдЪрдирд╛ред (c) рдПрдХ SimCSPSPPF рдмреНрд▓реЙрдХред (рдореВрд▓).

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

  • рджреНрд╡рд┐рджрд┐рд╢ рд╕рдВрдпреЛрдЬрди (рдмреАрдЖрдИрд╕реА) рдореЙрдбреНрдпреВрд▓: YOLOv6 рдбрд┐рдЯреЗрдХреНрдЯрд░ рдХреА рдЧрд░реНрджрди рдореЗрдВ рдПрдХ BiC рдореЙрдбреНрдпреВрд▓ рдкреЗрд╢ рдХрд░рддрд╛ рд╣реИ, рд╕реНрдерд╛рдиреАрдпрдХрд░рдг рд╕рдВрдХреЗрддреЛрдВ рдХреЛ рдмрдврд╝рд╛рддрд╛ рд╣реИ рдФрд░ рдирдЧрдгреНрдп рдЧрддрд┐ рдЧрд┐рд░рд╛рд╡рдЯ рдХреЗ рд╕рд╛рде рдкреНрд░рджрд░реНрд╢рди рд▓рд╛рдн рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред
  • рдПрдВрдХрд░-рдПрдбреЗрдб рдЯреНрд░реЗрдирд┐рдВрдЧ (AAT) рд░рдгрдиреАрддрд┐: рдпрд╣ рдореЙрдбрд▓ рдЕрдиреБрдорд╛рди рджрдХреНрд╖рддрд╛ рд╕реЗ рд╕рдордЭреМрддрд╛ рдХрд┐рдП рдмрд┐рдирд╛ рдПрдВрдХрд░-рдЖрдзрд╛рд░рд┐рдд рдФрд░ рдПрдВрдХрд░-рдореБрдХреНрдд рджреЛрдиреЛрдВ рдкреНрд░рддрд┐рдорд╛рдиреЛрдВ рдХреЗ рд▓рд╛рднреЛрдВ рдХрд╛ рдЖрдирдВрдж рд▓реЗрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдПрдЯреА рдХрд╛ рдкреНрд░рд╕реНрддрд╛рд╡ рдХрд░рддрд╛ рд╣реИред
  • рдмрдврд╝реА рд╣реБрдИ рд░реАрдврд╝ рдФрд░ рдЧрд░реНрджрди рдбрд┐рдЬрд╛рдЗрди: рд░реАрдврд╝ рдФрд░ рдЧрд░реНрджрди рдореЗрдВ рдПрдХ рдФрд░ рдЪрд░рдг рдХреЛ рд╢рд╛рдорд┐рд▓ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП YOLOv6 рдХреЛ рдЧрд╣рд░рд╛ рдХрд░рдХреЗ, рдпрд╣ рдореЙрдбрд▓ рдЙрдЪреНрдЪ-рд░рд┐рдЬрд╝реЙрд▓реНрдпреВрд╢рди рдЗрдирдкреБрдЯ рдкрд░ COCO рдбреЗрдЯрд╛рд╕реЗрдЯ рдкрд░ рдЕрддреНрдпрд╛рдзреБрдирд┐рдХ рдкреНрд░рджрд░реНрд╢рди рдкреНрд░рд╛рдкреНрдд рдХрд░рддрд╛ рд╣реИред
  • рд╕реНрд╡-рдЖрд╕рд╡рди рд░рдгрдиреАрддрд┐: YOLOv6 рдХреЗ рдЫреЛрдЯреЗ рдореЙрдбрд▓реЛрдВ рдХреЗ рдкреНрд░рджрд░реНрд╢рди рдХреЛ рдмрдврд╝рд╛рд╡рд╛ рджреЗрдиреЗ рдХреЗ рд▓рд┐рдП рдПрдХ рдирдИ рд╕реНрд╡-рдЖрд╕рд╡рди рд░рдгрдиреАрддрд┐ рд▓рд╛рдЧреВ рдХреА рдЧрдИ рд╣реИ, рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЗ рджреМрд░рд╛рди рд╕рд╣рд╛рдпрдХ рдкреНрд░рддрд┐рдЧрдорди рд╢рд╛рдЦрд╛ рдХреЛ рдмрдврд╝рд╛рдпрд╛ рдЧрдпрд╛ рд╣реИ рдФрд░ рдПрдХ рдЪрд┐рд╣реНрдирд┐рдд рдЧрддрд┐ рдореЗрдВ рдЧрд┐рд░рд╛рд╡рдЯ рд╕реЗ рдмрдЪрдиреЗ рдХреЗ рд▓рд┐рдП рдЗрд╕реЗ рдЕрдиреБрдорд╛рди рдкрд░ рд╣рдЯрд╛ рджрд┐рдпрд╛ рдЧрдпрд╛ рд╣реИред

рдкреНрд░рджрд░реНрд╢рди рдореЗрдЯреНрд░рд┐рдХреНрд╕

YOLOv6 рд╡рд┐рднрд┐рдиреНрди рдкреИрдорд╛рдиреЛрдВ рдХреЗ рд╕рд╛рде рд╡рд┐рднрд┐рдиреНрди рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓ рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИ:

  • YOLOv6-N: NVIDIA Tesla T37.5 GPU рдХреЗ рд╕рд╛рде 1187 FPS рдкрд░ COCO val2017 рдкрд░ 4% APред
  • YOLOv6-S: 484 FPS рдкрд░ 45.0% APред
  • YOLOv6-M: 50.0 FPS рдкрд░ 226% APред
  • YOLOv6-L: 52.8% AP 116 FPS рдкрд░ред
  • YOLOv6-L6: рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдореЗрдВ рдЕрддреНрдпрд╛рдзреБрдирд┐рдХ рд╕рдЯреАрдХрддрд╛ред

YOLOv6 рдореЛрдмрд╛рдЗрд▓ рдкреНрд▓реЗрдЯрдлреЙрд░реНрдо рдХреЗ рд▓рд┐рдП рдЕрдиреБрдХреВрд▓рд┐рдд рд╡рд┐рднрд┐рдиреНрди рдкрд░рд┐рд╢реБрджреНрдзрддрд╛ рдФрд░ рдореЙрдбрд▓ рдХреЗ рд▓рд┐рдП рдорд╛рддреНрд░рд╛рддреНрдордХ рдореЙрдбрд▓ рднреА рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред

рдЙрдкрдпреЛрдЧ рдХреЗ рдЙрджрд╛рд╣рд░рдг

рдпрд╣ рдЙрджрд╛рд╣рд░рдг рд╕рд░рд▓ YOLOv6 рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдФрд░ рдЕрдиреБрдорд╛рди рдЙрджрд╛рд╣рд░рдг рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред рдЗрди рдФрд░ рдЕрдиреНрдп рдореЛрдб рдкрд░ рдкреВрд░реНрдг рдкреНрд░рд▓реЗрдЦрди рдХреЗ рд▓рд┐рдП рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгреА, рдЯреНрд░реЗрди, рд╡реИрд▓ рдФрд░ рдирд┐рд░реНрдпрд╛рдд рдбреЙрдХреНрд╕ рдкреГрд╖реНрда рджреЗрдЦреЗрдВред

рдЙрджрд╛рд╣рд░рдг

PyTorch рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд *.pt рдореЙрдбрд▓ рдХреЗ рд╕рд╛рде-рд╕рд╛рде рдХреЙрдиреНрдлрд╝рд┐рдЧрд░реЗрд╢рди *.yaml рдлрд╝рд╛рдЗрд▓реЛрдВ рдХреЛ рдкрд╛рд╕ рдХрд┐рдпрд╛ рдЬрд╛ рд╕рдХрддрд╛ рд╣реИ YOLO() рдореЗрдВ рдПрдХ рдореЙрдбрд▓ рдЙрджрд╛рд╣рд░рдг рдмрдирд╛рдиреЗ рдХреЗ рд▓рд┐рдП рд╡рд░реНрдЧ python:

from ultralytics import YOLO

# Build a YOLOv6n model from scratch
model = YOLO('yolov6n.yaml')

# Display model information (optional)
model.info()

# Train the model on the COCO8 example dataset for 100 epochs
results = model.train(data='coco8.yaml', epochs=100, imgsz=640)

# Run inference with the YOLOv6n model on the 'bus.jpg' image
results = model('path/to/bus.jpg')

CLI рдореЙрдбрд▓ рдХреЛ рд╕реАрдзреЗ рдЪрд▓рд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдХрдорд╛рдВрдб рдЙрдкрд▓рдмреНрдз рд╣реИрдВ:

# Build a YOLOv6n model from scratch and train it on the COCO8 example dataset for 100 epochs
yolo train model=yolov6n.yaml data=coco8.yaml epochs=100 imgsz=640

# Build a YOLOv6n model from scratch and run inference on the 'bus.jpg' image
yolo predict model=yolov6n.yaml source=path/to/bus.jpg

рд╕рдорд░реНрдерд┐рдд рдХрд╛рд░реНрдп рдФрд░ рдореЛрдб

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

рдореЙрдбрд▓ рдкреНрд░рдХрд╛рд░ рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рд╡рдЬрди рд╕рдорд░реНрдерд┐рдд рдХрд╛рд░реНрдп рдЕрдиреБрдорд╛рди рдорд╛рдиреНрдпрддрд╛ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдирд┐рд░реНрдпрд╛рддрд┐рдд рдорд╛рд▓
рдпреЛрд▓реЛрд╡6-рдПрди yolov6-n.pt рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ тЬЕ тЬЕ тЬЕ тЬЕ
рдпреЛрд▓реЛрд╡6-рдПрд╕ yolov6-s.pt рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ тЬЕ тЬЕ тЬЕ тЬЕ
рдпреЛрд▓реЛрд╡6-рдПрдо yolov6-m.pt рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ тЬЕ тЬЕ тЬЕ тЬЕ
рдпреЛрд▓реЛрд╡6-рдПрд▓ yolov6-l.pt рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ тЬЕ тЬЕ тЬЕ тЬЕ
рдпреЛрд▓реЛрд╡6-рдПрд▓6 yolov6-l6.pt рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ тЬЕ тЬЕ тЬЕ тЬЕ

рдпрд╣ рддрд╛рд▓рд┐рдХрд╛ YOLOv6 рдореЙрдбрд▓ рд╡реЗрд░рд┐рдПрдВрдЯ рдХрд╛ рд╡рд┐рд╕реНрддреГрдд рдЕрд╡рд▓реЛрдХрди рдкреНрд░рджрд╛рди рдХрд░рддреА рд╣реИ, рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХрд╛рд░реНрдпреЛрдВ рдореЗрдВ рдЙрдирдХреА рдХреНрд╖рдорддрд╛рдУрдВ рдФрд░ рдЕрдиреБрдорд╛рди, рд╕рддреНрдпрд╛рдкрди, рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдФрд░ рдирд┐рд░реНрдпрд╛рдд рдЬреИрд╕реЗ рд╡рд┐рднрд┐рдиреНрди рдкрд░рд┐рдЪрд╛рд▓рди рдореЛрдб рдХреЗ рд╕рд╛рде рдЙрдирдХреА рд╕рдВрдЧрддрддрд╛ рдХреЛ рдЙрдЬрд╛рдЧрд░ рдХрд░рддреА рд╣реИред рдпрд╣ рд╡реНрдпрд╛рдкрдХ рд╕рдорд░реНрдерди рд╕реБрдирд┐рд╢реНрдЪрд┐рдд рдХрд░рддрд╛ рд╣реИ рдХрд┐ рдЙрдкрдпреЛрдЧрдХрд░реНрддрд╛ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдкрд░рд┐рджреГрд╢реНрдпреЛрдВ рдХреА рдПрдХ рд╡рд┐рд╕реНрддреГрдд рд╢реНрд░реГрдВрдЦрд▓рд╛ рдореЗрдВ YOLOv6 рдореЙрдбрд▓ рдХреА рдХреНрд╖рдорддрд╛рдУрдВ рдХрд╛ рдкреВрд░реА рддрд░рд╣ рд╕реЗ рд▓рд╛рдн рдЙрдард╛ рд╕рдХрддреЗ рд╣реИрдВред

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

рд╣рдо рд▓реЗрдЦрдХреЛрдВ рдХреЛ рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рдХреНрд╖реЗрддреНрд░ рдореЗрдВ рдЙрдирдХреЗ рдорд╣рддреНрд╡рдкреВрд░реНрдг рдпреЛрдЧрджрд╛рди рдХреЗ рд▓рд┐рдП рд╕реНрд╡реАрдХрд╛рд░ рдХрд░рдирд╛ рдЪрд╛рд╣рддреЗ рд╣реИрдВ:

@misc{li2023yolov6,
      title={YOLOv6 v3.0: A Full-Scale Reloading},
      author={Chuyi Li and Lulu Li and Yifei Geng and Hongliang Jiang and Meng Cheng and Bo Zhang and Zaidan Ke and Xiaoming Xu and Xiangxiang Chu},
      year={2023},
      eprint={2301.05586},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

рдореВрд▓ YOLOv6 рдкреЗрдкрд░ arXiv рдкрд░ рдкрд╛рдпрд╛ рдЬрд╛ рд╕рдХрддрд╛ рд╣реИред рд▓реЗрдЦрдХреЛрдВ рдиреЗ рдЕрдкрдирд╛ рдХрд╛рдо рд╕рд╛рд░реНрд╡рдЬрдирд┐рдХ рд░реВрдк рд╕реЗ рдЙрдкрд▓рдмреНрдз рдХрд░рд╛рдпрд╛ рд╣реИ, рдФрд░ рдХреЛрдбрдмреЗрд╕ рдХреЛ GitHub рдкрд░ рдПрдХреНрд╕реЗрд╕ рдХрд┐рдпрд╛ рдЬрд╛ рд╕рдХрддрд╛ рд╣реИред рд╣рдо рдХреНрд╖реЗрддреНрд░ рдХреЛ рдЖрдЧреЗ рдмрдврд╝рд╛рдиреЗ рдФрд░ рдЙрдирдХреЗ рдХрд╛рдо рдХреЛ рд╡реНрдпрд╛рдкрдХ рд╕рдореБрджрд╛рдп рдХреЗ рд▓рд┐рдП рд╕реБрд▓рдн рдмрдирд╛рдиреЗ рдореЗрдВ рдЙрдирдХреЗ рдкреНрд░рдпрд╛рд╕реЛрдВ рдХреА рд╕рд░рд╛рд╣рдирд╛ рдХрд░рддреЗ рд╣реИрдВред



2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-01-07
рд▓реЗрдЦрдХ: рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (5)

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