YOLOv5
рд╡рд┐рд╣рдВрдЧрд╛рд╡рд▓реЛрдХрди
YOLOv5u рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдкрджреНрдзрддрд┐рдпреЛрдВ рдореЗрдВ рдкреНрд░рдЧрддрд┐ рдХрд╛ рдкреНрд░рддрд┐рдирд┐рдзрд┐рддреНрд╡ рдХрд░рддрд╛ рд╣реИред рдХреА рдореВрд▓рднреВрдд рд╡рд╛рд╕реНрддреБрдХрд▓рд╛ рд╕реЗ рдЙрддреНрдкрдиреНрди YOLOv5 рджреНрд╡рд╛рд░рд╛ рд╡рд┐рдХрд╕рд┐рдд рдореЙрдбрд▓ Ultralytics, YOLOv5u рдПрдВрдХрд░-рдореБрдХреНрдд, рдСрдмреНрдЬреЗрдХреНрдЯрдиреЗрд╕-рдлреНрд░реА рд╕реНрдкреНрд▓рд┐рдЯ рд╣реЗрдб рдХреЛ рдПрдХреАрдХреГрдд рдХрд░рддрд╛ рд╣реИ, рдЬреЛ рдкрд╣рд▓реЗ рдореЗрдВ рдкреЗрд╢ рдХрд┐рдпрд╛ рдЧрдпрд╛ рдПрдХ рдлреАрдЪрд░ рдерд╛ YOLOv8 рдореЙрдбрд▓ред рдпрд╣ рдЕрдиреБрдХреВрд▓рди рдореЙрдбрд▓ рдХреА рд╡рд╛рд╕реНрддреБрдХрд▓рд╛ рдХреЛ рдкрд░рд┐рд╖реНрдХреГрдд рдХрд░рддрд╛ рд╣реИ, рдЬрд┐рд╕рд╕реЗ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХрд╛рд░реНрдпреЛрдВ рдореЗрдВ рдмреЗрд╣рддрд░ рд╕рдЯреАрдХрддрд╛-рдЧрддрд┐ рдЯреНрд░реЗрдбрдСрдл рд╣реЛрддрд╛ рд╣реИред рдЕрдиреБрднрд╡рдЬрдиреНрдп рдкрд░рд┐рдгрд╛рдореЛрдВ рдФрд░ рдЗрд╕рдХреА рд╡реНрдпреБрддреНрдкрдиреНрди рд╡рд┐рд╢реЗрд╖рддрд╛рдУрдВ рдХреЛ рджреЗрдЦрддреЗ рд╣реБрдП, YOLOv5u рдЕрдиреБрд╕рдВрдзрд╛рди рдФрд░ рд╡реНрдпрд╛рд╡рд╣рд╛рд░рд┐рдХ рдЕрдиреБрдкреНрд░рдпреЛрдЧреЛрдВ рджреЛрдиреЛрдВ рдореЗрдВ рдордЬрдмреВрдд рд╕рдорд╛рдзрд╛рди рдЪрд╛рд╣рдиреЗ рд╡рд╛рд▓реЛрдВ рдХреЗ рд▓рд┐рдП рдПрдХ рдХреБрд╢рд▓ рд╡рд┐рдХрд▓реНрдк рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред
рдкреНрд░рдореБрдЦ рд╡рд┐рд╢реЗрд╖рддрд╛рдРрдВ
-
рдПрдВрдХрд░-рдлреНрд░реА рд╕реНрдкреНрд▓рд┐рдЯ 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 рдореЙрдбрд▓ рдХреЛ рд╕реАрдзреЗ рдЪрд▓рд╛рдиреЗ рдХреЗ рд▓рд┐рдП рдХрдорд╛рдВрдб рдЙрдкрд▓рдмреНрдз рд╣реИрдВ:
рдкреНрд░рд╢рдВрд╕рд╛ рдкрддреНрд░ рдФрд░ рдкрд╛рд╡рддреА
рдпрджрд┐ рдЖрдк рдЙрдкрдпреЛрдЧ рдХрд░рддреЗ рд╣реИрдВ YOLOv5 рдпрд╛ YOLOv5u рдЕрдкрдиреЗ рд╢реЛрдз рдореЗрдВ, рдХреГрдкрдпрд╛ рдЙрджреНрдзреГрдд рдХрд░реЗрдВ Ultralytics YOLOv5 рднрдВрдбрд╛рд░ рдирд┐рдореНрдирд╛рдиреБрд╕рд╛рд░ рд╣реИ:
рдХреГрдкрдпрд╛ рдзреНрдпрд╛рди рджреЗрдВ рдХрд┐ YOLOv5 рдореЙрдбрд▓ рдХреЗ рддрд╣рдд рдкреНрд░рджрд╛рди рдХреА рдЬрд╛рддреА рд╣реИрдВ AGPL-3.0 рдФрд░ рдПрдВрдЯрд░рдкреНрд░рд╛рдЗрдЬрд╝ рд▓рд╛рдЗрд╕реЗрдВрд╕ред
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-04-17
рд▓реЗрдЦрдХ: рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (9)