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

YOLOv5

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

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

Ultralytics YOLOv5

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

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

  • рдЕрдиреБрдХреВрд▓рд┐рдд рд╕рдЯреАрдХрддрд╛-рдЧрддрд┐ рдЯреНрд░реЗрдбрдСрдлрд╝: рдЧрддрд┐ рдФрд░ рд╕рдЯреАрдХрддрд╛ рдЕрдХреНрд╕рд░ рд╡рд┐рдкрд░реАрдд рджрд┐рд╢рд╛рдУрдВ рдореЗрдВ рдЦреАрдВрдЪрддреА рд╣реИред рд▓реЗрдХрд┐рди YOLOv5u рдЗрд╕ рдЯреНрд░реЗрдбрдСрдлрд╝ рдХреЛ рдЪреБрдиреМрддреА рджреЗрддрд╛ рд╣реИред рдпрд╣ рдПрдХ рдХреИрд▓рд┐рдмреНрд░реЗрдЯреЗрдб рд╕рдВрддреБрд▓рди рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИ, рд╕рдЯреАрдХрддрд╛ рд╕реЗ рд╕рдордЭреМрддрд╛ рдХрд┐рдП рдмрд┐рдирд╛ рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЛ рд╕реБрдирд┐рд╢реНрдЪрд┐рдд рдХрд░рддрд╛ рд╣реИред рдпрд╣ рд╕реБрд╡рд┐рдзрд╛ рдЙрди рдЕрдиреБрдкреНрд░рдпреЛрдЧреЛрдВ рдХреЗ рд▓рд┐рдП рд╡рд┐рд╢реЗрд╖ рд░реВрдк рд╕реЗ рдЕрдореВрд▓реНрдп рд╣реИ рдЬреЛ рд╕реНрд╡рд╛рдпрддреНрдд рд╡рд╛рд╣рдиреЛрдВ, рд░реЛрдмреЛрдЯрд┐рдХреНрд╕ рдФрд░ рд░реАрдпрд▓-рдЯрд╛рдЗрдо рд╡реАрдбрд┐рдпреЛ рдПрдирд╛рд▓рд┐рдЯрд┐рдХреНрд╕ рдЬреИрд╕реЗ рддреНрд╡рд░рд┐рдд рдкреНрд░рддрд┐рдХреНрд░рд┐рдпрд╛рдУрдВ рдХреА рдорд╛рдВрдЧ рдХрд░рддреЗ рд╣реИрдВред

  • рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓ рдХреА рд╡рд┐рд╡рд┐рдзрддрд╛: рдпрд╣ рд╕рдордЭрддреЗ рд╣реБрдП рдХрд┐ рд╡рд┐рднрд┐рдиреНрди рдХрд╛рд░реНрдпреЛрдВ рдХреЗ рд▓рд┐рдП рдЕрд▓рдЧ-рдЕрд▓рдЧ рдЯреВрд▓рд╕реЗрдЯ рдХреА рдЖрд╡рд╢реНрдпрдХрддрд╛ рд╣реЛрддреА рд╣реИ, YOLOv5u рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓ рдХреА рдЕрдзрд┐рдХрддрд╛ рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред рдЪрд╛рд╣реЗ рдЖрдк рдЕрдиреБрдорд╛рди, рд╕рддреНрдпрд╛рдкрди, рдпрд╛ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдкрд░ рдзреНрдпрд╛рди рдХреЗрдВрджреНрд░рд┐рдд рдХрд░ рд░рд╣реЗ рд╣реЛрдВ, рдПрдХ рджрд░реНрдЬреА рдореЙрдбрд▓ рдЖрдкрдХреА рдкреНрд░рддреАрдХреНрд╖рд╛ рдХрд░ рд░рд╣рд╛ рд╣реИред рдпрд╣ рд╡рд┐рд╡рд┐рдзрддрд╛ рд╕реБрдирд┐рд╢реНрдЪрд┐рдд рдХрд░рддреА рд╣реИ рдХрд┐ рдЖрдк рдХреЗрд╡рд▓ рдПрдХ рдЖрдХрд╛рд░-рдлрд┐рдЯ-рд╕рднреА рд╕рдорд╛рдзрд╛рди рдХрд╛ рдЙрдкрдпреЛрдЧ рдирд╣реАрдВ рдХрд░ рд░рд╣реЗ рд╣реИрдВ, рдмрд▓реНрдХрд┐ рдПрдХ рдореЙрдбрд▓ рд╡рд┐рд╢реЗрд╖ рд░реВрдк рд╕реЗ рдЖрдкрдХреА рдЕрдиреВрдареА рдЪреБрдиреМрддреА рдХреЗ рд▓рд┐рдП рдареАрдХ-рдард╛рдХ рд╣реИред

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

YOLOv5u рдореЙрдбрд▓, рд╡рд┐рднрд┐рдиреНрди рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рд╡рдЬрди рдХреЗ рд╕рд╛рде, рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХрд╛рд░реНрдпреЛрдВ рдореЗрдВ рдЙрддреНрдХреГрд╖реНрдЯрддрд╛ рдкреНрд░рд╛рдкреНрдд рдХрд░рддреЗ рд╣реИрдВред рд╡реЗ рдореЛрдб рдХреА рдПрдХ рд╡рд┐рд╕реНрддреГрдд рд╢реНрд░реГрдВрдЦрд▓рд╛ рдХрд╛ рд╕рдорд░реНрдерди рдХрд░рддреЗ рд╣реИрдВ, рдЬреЛ рдЙрдиреНрд╣реЗрдВ рд╡рд┐рдХрд╛рд╕ рд╕реЗ рд▓реЗрдХрд░ рддреИрдирд╛рддреА рддрдХ рд╡рд┐рд╡рд┐рдз рдЕрдиреБрдкреНрд░рдпреЛрдЧреЛрдВ рдХреЗ рд▓рд┐рдП рдЙрдкрдпреБрдХреНрдд рдмрдирд╛рддреЗ рд╣реИрдВред

рдореЙрдбрд▓ рдкреНрд░рдХрд╛рд░ рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рд╡рдЬрди рдирд┐рдпрдд рдХрд╛рд░реНрдп рдЕрдиреБрдорд╛рди рдорд╛рдиреНрдпрддрд╛ рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдирд┐рд░реНрдпрд╛рддрд┐рдд рдорд╛рд▓
рдпреЛрд▓реЛрд╡5рдпреВ yolov5nu, yolov5su, yolov5mu, yolov5lu, yolov5xu, yolov5n6u, yolov5s6u, yolov5m6u, yolov5l6u, yolov5x6u рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ тЬЕ тЬЕ тЬЕ тЬЕ

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

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

рдкреНрд░рджрд░реНрд╢рди

COCO рдкрд░ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдЗрди рдореЙрдбрд▓реЛрдВ рдХреЗ рд╕рд╛рде рдЙрдкрдпреЛрдЧ рдХреЗ рдЙрджрд╛рд╣рд░рдгреЛрдВ рдХреЗ рд▓рд┐рдП рдбрд┐рдЯреЗрдХреНрд╢рди рдбреЙрдХреНрд╕ рджреЗрдЦреЗрдВ, рдЬрд┐рд╕рдореЗрдВ 80 рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдХрдХреНрд╖рд╛рдПрдВ рд╢рд╛рдорд┐рд▓ рд╣реИрдВред

рдХреЛ рдЧрдврд╝рдирд╛ рд╡рд╛рдИрдПрдПрдордПрд▓ рдЖрдХрд╛рд░ рд╡рд╛рд▓рд╛
(рдкрд┐рдХреНрд╕реЗрд▓)
рдорд╛рдирдЪрд┐рддреНрд░рд╡реИрд▓
50-95
рдЧрддрд┐
рд╕реАрдкреАрдпреВ ONNX
(рдПрдордПрд╕)
рдЧрддрд┐
рдП100 TensorRT
(рдПрдордПрд╕)
рдкрд░рдо
(рдПрдо)
рдлреНрд▓реЙрдк
(рдмреА)
yolov5nu.pt yolov5n.yaml 640 34.3 73.6 1.06 2.6 7.7
yolov5su.pt yolov5s.yaml 640 43.0 120.7 1.27 9.1 24.0
yolov5mu.pt yolov5m.yaml 640 49.0 233.9 1.86 25.1 64.2
yolov5lu.pt yolov5l.yaml 640 52.2 408.4 2.50 53.2 135.0
yolov5xu.pt yolov5x.yaml 640 53.2 763.2 3.81 97.2 246.4
yolov5n6u.pt рдпреЛрд▓реЛрд╡5рдПрди6.рдпрдорд▓ 1280 42.1 211.0 1.83 4.3 7.8
yolov5s6u.pt рдпреЛрд▓реЛрд╡5рдПрд╕6.рдпрдореНрд▓ 1280 48.6 422.6 2.34 15.3 24.6
yolov5m6u.pt yolov5m6.yaml 1280 53.6 810.9 4.36 41.2 65.7
yolov5l6u.pt yolov5l6.yaml 1280 55.7 1470.9 5.47 86.1 137.4
yolov5x6u.pt рдпреЛрд▓реЛрд╡5x6.рдпрдорд▓ 1280 56.8 2436.5 8.98 155.4 250.7

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

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

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

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

from ultralytics import YOLO

# Load a COCO-pretrained YOLOv5n model
model = YOLO('yolov5n.pt')

# 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 YOLOv5n model on the 'bus.jpg' image
results = model('path/to/bus.jpg')

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

# Load a COCO-pretrained YOLOv5n model and train it on the COCO8 example dataset for 100 epochs
yolo train model=yolov5n.pt data=coco8.yaml epochs=100 imgsz=640

# Load a COCO-pretrained YOLOv5n model and run inference on the 'bus.jpg' image
yolo predict model=yolov5n.pt source=path/to/bus.jpg

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

рдпрджрд┐ рдЖрдк рдЙрдкрдпреЛрдЧ рдХрд░рддреЗ рд╣реИрдВ YOLOv5 рдпрд╛ YOLOv5u рдЕрдкрдиреЗ рд╢реЛрдз рдореЗрдВ, рдХреГрдкрдпрд╛ рдЙрджреНрдзреГрдд рдХрд░реЗрдВ Ultralytics YOLOv5 рднрдВрдбрд╛рд░ рдирд┐рдореНрдирд╛рдиреБрд╕рд╛рд░ рд╣реИ:

@software{yolov5,
  title = {Ultralytics YOLOv5},
  author = {Glenn Jocher},
  year = {2020},
  version = {7.0},
  license = {AGPL-3.0},
  url = {https://github.com/ultralytics/yolov5},
  doi = {10.5281/zenodo.3908559},
  orcid = {0000-0001-5950-6979}
}

рдХреГрдкрдпрд╛ рдзреНрдпрд╛рди рджреЗрдВ рдХрд┐ YOLOv5 рдореЙрдбрд▓ рдХреЗ рддрд╣рдд рдкреНрд░рджрд╛рди рдХреА рдЬрд╛рддреА рд╣реИрдВ AGPL-3.0 рдФрд░ рдПрдВрдЯрд░рдкреНрд░рд╛рдЗрдЬрд╝ рд▓рд╛рдЗрд╕реЗрдВрд╕ред



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

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