YOLOv5 рдХреНрд╡рд┐рдХрд╕реНрдЯрд╛рд░реНрдЯ ЁЯЪА
рдХреЗ рд╕рд╛рде рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рдЧрддрд┐рд╢реАрд▓ рджрд╛рдпрд░реЗ рдореЗрдВ рдЕрдкрдиреА рдпрд╛рддреНрд░рд╛ рдкрд░ рд▓рдЧрдирд╛ YOLOv5! рдпрд╣ рдорд╛рд░реНрдЧрджрд░реНрд╢рд┐рдХрд╛ рдПрдЖрдИ рдЙрддреНрд╕рд╛рд╣реА рдФрд░ рдкреЗрд╢реЗрд╡рд░реЛрдВ рдХреЗ рд▓рд┐рдП рдПрдХ рд╡реНрдпрд╛рдкрдХ рдкреНрд░рд╛рд░рдВрднрд┐рдХ рдмрд┐рдВрджреБ рдХреЗ рд░реВрдк рдореЗрдВ рд╕реЗрд╡рд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рддреИрдпрд╛рд░ рдХреА рдЧрдИ рд╣реИ рдЬреЛ рдорд╛рд╕реНрдЯрд░ рдХрд░рдиреЗ рдХрд╛ рд▓рдХреНрд╖реНрдп рд░рдЦрддреЗ рд╣реИрдВ YOLOv5. рдкреНрд░рд╛рд░рдВрднрд┐рдХ рд╕реЗрдЯрдЕрдк рд╕реЗ рд▓реЗрдХрд░ рдЙрдиреНрдирдд рдкреНрд░рд╢рд┐рдХреНрд╖рдг рддрдХрдиреАрдХреЛрдВ рддрдХ, рд╣рдордиреЗ рдЖрдкрдХреЛ рдХрд╡рд░ рдХрд┐рдпрд╛ рд╣реИред рдЗрд╕ рдЧрд╛рдЗрдб рдХреЗ рдЕрдВрдд рддрдХ, рдЖрдкрдХреЗ рдкрд╛рд╕ рд▓рд╛рдЧреВ рдХрд░рдиреЗ рдХрд╛ рдЬреНрдЮрд╛рди рд╣реЛрдЧрд╛ YOLOv5 рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╕реЗ рдЕрдкрдиреА рдкрд░рд┐рдпреЛрдЬрдирд╛рдУрдВ рдореЗрдВред рдЖрдЗрдП рдЗрдВрдЬрдиреЛрдВ рдХреЛ рдкреНрд░рдЬреНрд╡рд▓рд┐рдд рдХрд░реЗрдВ рдФрд░ рдЪрдврд╝реЗрдВ YOLOv5!
рдкрджрд╛рд╕реАрди рдХрд░рдирд╛
рд░рд┐рдкреЙрдЬрд┐рдЯрд░реА рдХреА рдХреНрд▓реЛрдирд┐рдВрдЧ рдФрд░ рдкрд░реНрдпрд╛рд╡рд░рдг рдХреА рд╕реНрдерд╛рдкрдирд╛ рдХрд░рдХреЗ рд▓реЙрдиреНрдЪ рдХреА рддреИрдпрд╛рд░реА рдХрд░реЗрдВред рдпрд╣ рд╕реБрдирд┐рд╢реНрдЪрд┐рдд рдХрд░рддрд╛ рд╣реИ рдХрд┐ рд╕рднреА рдЖрд╡рд╢реНрдпрдХ рдЖрд╡рд╢реНрдпрдХрддрд╛рдПрдВ рд╕реНрдерд╛рдкрд┐рдд рд╣реИрдВред рдЬрд╛рдВрдЪреЗрдВ рдХрд┐ рдЖрдкрдХреЗ рдкрд╛рд╕ рд╣реИ Python>=3.8.0 рдФрд░ PyTorch> = 1.8 рдЯреЗрдХрдСрдлрд╝ рдХреЗ рд▓рд┐рдП рддреИрдпрд╛рд░ред
git clone https://github.com/ultralytics/yolov5 # clone repository
cd yolov5
pip install -r requirements.txt # install dependencies
рдХреЗ рд╕рд╛рде рдЕрдиреБрдорд╛рди PyTorch рдЪрдХреНрд░рдирд╛рднрд┐
рдХреА рд╕рд╛рджрдЧреА рдХрд╛ рдЕрдиреБрднрд╡ рдХрд░реЗрдВ YOLOv5 PyTorch рд╣рдм рдЕрдиреБрдорд╛рди, рдЬрд╣рд╛рдВ рдореЙрдбрд▓ рдирд╡реАрдирддрдо рд╕реЗ рдореВрд▓ рд░реВрдк рд╕реЗ рдбрд╛рдЙрдирд▓реЛрдб рдХрд┐рдП рдЬрд╛рддреЗ рд╣реИрдВ YOLOv5 рд░рд┐рд▓реАрдЬред
import torch
# Model loading
model = torch.hub.load("ultralytics/yolov5", "yolov5s") # Can be 'yolov5n' - 'yolov5x6', or 'custom'
# Inference on images
img = "https://ultralytics.com/images/zidane.jpg" # Can be a file, Path, PIL, OpenCV, numpy, or list of images
# Run inference
results = model(img)
# Display results
results.print() # Other options: .show(), .save(), .crop(), .pandas(), etc.
detect.py рдХреЗ рд╕рд╛рде рдЕрдиреБрдорд╛рди
рдЙрдкрд╕реНрдХрд░рдг detect.py
рд╡рд┐рднрд┐рдиреНрди рд╕реНрд░реЛрддреЛрдВ рдкрд░ рдмрд╣реБрдореБрдЦреА рдЕрдиреБрдорд╛рди рдХреЗ рд▓рд┐рдПред рдпрд╣ рд╕реНрд╡рдЪрд╛рд▓рд┐рдд рд░реВрдк рд╕реЗ рдкреНрд░рд╛рдкреНрдд рдХрд░рддрд╛ рд╣реИ рдореЙрдбрд▓ рдирд╡реАрдирддрдо рд╕реЗ YOLOv5 рдирд┐рд░реНрдЧрдорди рдФрд░ рдЖрд╕рд╛рдиреА рд╕реЗ рдкрд░рд┐рдгрд╛рдо рдмрдЪрд╛рддрд╛ рд╣реИред
python detect.py --weights yolov5s.pt --source 0 # webcam
img.jpg # image
vid.mp4 # video
screen # screenshot
path/ # directory
list.txt # list of images
list.streams # list of streams
'path/*.jpg' # glob
'https://youtu.be/LNwODJXcvt4' # YouTube
'rtsp://example.com/media.mp4' # RTSP, RTMP, HTTP stream
рдкреНрд░рд╢рд┐рдХреНрд╖рдг
рдХреЛ рджреЛрд╣рд░рд╛рдирд╛ YOLOv5 рдХреЛрдХреЛ рдиреАрдЪреЗ рджрд┐рдП рдЧрдП рдирд┐рд░реНрджреЗрд╢реЛрдВ рдХреЗ рд╕рд╛рде рдмреЗрдВрдЪрдорд╛рд░реНрдХред рдЖрд╡рд╢реНрдпрдХ рдореЙрдбрд▓ рдФрд░ рдбреЗрдЯрд╛рд╕реЗрдЯ рдирд╡реАрдирддрдо рд╕реЗ рд╕реАрдзреЗ рдЦреАрдВрдЪреЗ рдЬрд╛рддреЗ рд╣реИрдВ YOLOv5 рдирд┐рд░реНрдЧрдорди. V5 GPU рдкрд░ YOLOv100n/s/m/l/x рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдореЗрдВ рдЖрдорддреМрд░ рдкрд░ рдХреНрд░рдорд╢рдГ 1/2/4/6/8 рджрд┐рди рд▓рдЧрдиреЗ рдЪрд╛рд╣рд┐рдП (рдзреНрдпрд╛рди рджреЗрдВ рдХрд┐ рдорд▓реНрдЯреА-рдЬреАрдкреАрдпреВ рд╕реЗрдЯрдЕрдк рддреЗрдЬреА рд╕реЗ рдХрд╛рдо рдХрд░рддреЗ рд╣реИрдВ)ред рдЙрдЪреНрдЪрддрдо рд╕рдВрднрд╡ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдкреНрд░рджрд░реНрд╢рди рдХреЛ рдЕрдзрд┐рдХрддрдо рдХрд░реЗрдВ --batch-size
рдпрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВ --batch-size -1
рдХреЗ рд▓рд┐рдП YOLOv5 рдСрдЯреЛрдмреИрдЪ рд▓рдХреНрд╖рдгред рдирд┐рдореНрдирд▓рд┐рдЦрд┐рдд рдмреИрдЪ рдЖрдХрд╛рд░ V100-16GB GPU рдХреЗ рд▓рд┐рдП рдЖрджрд░реНрд╢ рд╣реИрдВред
python train.py --data coco.yaml --epochs 300 --weights '' --cfg yolov5n.yaml --batch-size 128
yolov5s 64
yolov5m 40
yolov5l 24
yolov5x 16
рдирд┐рд╖реНрдХрд░реНрд╖ рдирд┐рдХрд╛рд▓рдиреЗ рдХреЗ рд▓рд┐рдП, YOLOv5 рди рдХреЗрд╡рд▓ рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХреЗ рд▓рд┐рдП рдПрдХ рдЕрддреНрдпрд╛рдзреБрдирд┐рдХ рдЙрдкрдХрд░рдг рд╣реИ, рдмрд▓реНрдХрд┐ рджреГрд╢реНрдп рд╕рдордЭ рдХреЗ рдорд╛рдзреНрдпрдо рд╕реЗ рджреБрдирд┐рдпрд╛ рдХреЗ рд╕рд╛рде рдмрд╛рддрдЪреАрдд рдХрд░рдиреЗ рдХреЗ рддрд░реАрдХреЗ рдХреЛ рдмрджрд▓рдиреЗ рдореЗрдВ рдорд╢реАрди рд▓рд░реНрдирд┐рдВрдЧ рдХреА рд╢рдХреНрддрд┐ рдХрд╛ рдПрдХ рд╡рд╕реАрдпрддрдирд╛рдорд╛ рднреА рд╣реИред рдЬреИрд╕рд╛ рдХрд┐ рдЖрдк рдЗрд╕ рдЧрд╛рдЗрдб рдХреЗ рдорд╛рдзреНрдпрдо рд╕реЗ рдЖрдЧреЗ рдмрдврд╝рддреЗ рд╣реИрдВ рдФрд░ рдЖрд╡реЗрджрди рдХрд░рдирд╛ рд╢реБрд░реВ рдХрд░рддреЗ рд╣реИрдВ YOLOv5 рдЕрдкрдиреА рдкрд░рд┐рдпреЛрдЬрдирд╛рдУрдВ рдХреЗ рд▓рд┐рдП, рдпрд╛рдж рд░рдЦреЗрдВ рдХрд┐ рдЖрдк рдПрдХ рддрдХрдиреАрдХреА рдХреНрд░рд╛рдВрддрд┐ рдореЗрдВ рд╕рдмрд╕реЗ рдЖрдЧреЗ рд╣реИрдВ, рдЬреЛ рдЙрд▓реНрд▓реЗрдЦрдиреАрдп рдЙрдкрд▓рдмреНрдзрд┐ рд╣рд╛рд╕рд┐рд▓ рдХрд░рдиреЗ рдореЗрдВ рд╕рдХреНрд╖рдо рд╣реИрдВред рдХреНрдпрд╛ рдЖрдкрдХреЛ рд╕рд╛рдереА рджреВрд░рджрд░реНрд╢реА рд▓реЛрдЧреЛрдВ рд╕реЗ рдФрд░ рдЕрдВрддрд░реНрджреГрд╖реНрдЯрд┐ рдпрд╛ рд╕рдорд░реНрдерди рдХреА рдЖрд╡рд╢реНрдпрдХрддрд╛ рд╣реИ, рдЖрдкрдХреЛ рдбреЗрд╡рд▓рдкрд░реНрд╕ рдФрд░ рд╢реЛрдзрдХрд░реНрддрд╛рдУрдВ рдХреЗ рд╕рдВрдкрдиреНрди рд╕рдореБрджрд╛рдп рдХреЗ рд▓рд┐рдП рд╣рдорд╛рд░реЗ GitHub рд░рд┐рдкреЙрдЬрд┐рдЯрд░реА рд╣реЛрдо рдореЗрдВ рдЖрдордВрддреНрд░рд┐рдд рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИред рдЦреЛрдЬ рдХрд░рддреЗ рд░рд╣реЗрдВ, рдирд╡рд╛рдЪрд╛рд░ рдХрд░рддреЗ рд░рд╣реЗрдВ, рдФрд░ рдХреЗ рдЪрдорддреНрдХрд╛рд░реЛрдВ рдХрд╛ рдЖрдирдВрдж рд▓реЗрдВ YOLOv5. рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдореЗрдВ рдЦреБрд╢реА! ЁЯМаЁЯФН
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2023-12-03
рд▓реЗрдЦрдХ: рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (2)