自动驾驶－YOLOV5目标检测

1、环境：

操作系统：Ubuntu20.04

Python3.9

Pytorch1.10.2

IDE：（本人使用的是ECLIPSE）

显卡：NVIDIA GeForce GTX 1060 6GB

开源库地址：GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

2、配置参数

解压后，在ECLIPSE打开源码。

打开detect.py文件。

核心配置

    #模形设置，这里默认是小模型

IMG=640的四种模型

yolov5s.pt
yolov5m.pt
yolov5l.pt
yolov5x.pt

IMG＝1280的四种模型

yolov5s6.pt
yolov5m6.pt
yolov5l6.pt
yolov5x6.pt

    parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)')
    #要识别的源图片路径
    parser.add_argument('--source', type=str, default='