ThinkChat🤖让你学习和工作更高效,注册即送10W Token,即刻开启你的AI之旅 广告
[TOC] > [github](https://github.com/ultralytics/yolov5) ## 概述 YOLOv5(You Only Look Once version 5)是一种计算机视觉中的目标检测模型,它是YOLO(You Only Look Once)目标检测系列的第五个版本。YOLO是一种用于实时目标检测的先进算法,其主要特点是速度快、准确性高,以及能够在实时或嵌入式系统上运行 ## 安装 ``` git clone https://github.com/ultralytics/yolov5 # clone cd yolov5 pip install -r requirements.txt # install ``` ## 使用 可支持的资源类型 ``` 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 ``` 可支持的模型 ``` python detect.py --weights yolov5s.pt # PyTorch yolov5s.torchscript # TorchScript yolov5s.onnx # ONNX Runtime or OpenCV DNN with --dnn yolov5s_openvino_model # OpenVINO yolov5s.engine # TensorRT yolov5s.mlmodel # CoreML (macOS-only) yolov5s_saved_model # TensorFlow SavedModel yolov5s.pb # TensorFlow GraphDef yolov5s.tflite # TensorFlow Lite yolov5s_edgetpu.tflite # TensorFlow Edge TPU yolov5s_paddle_model # PaddlePaddle ``` ## 示例 ### 测试自带模型 如果 yolov5s.pt 不存在,为自动下载 ``` python detect.py --weights yolov5s.pt --source data/images/zidane.jpg ```