ubuntu18.04配置yolov5
时间: 2023-04-28 13:03:50 浏览: 314
YOLOv5(PyTorch)实战:训练自己的数据集(Ubuntu)
首先,需要确保电脑上已经安装了 CUDA 和 cuDNN。
然后,在终端中运行以下命令来安装必要的 Python 库:
```
pip install torch torchvision
pip install opencv-python
pip install -U torchvision
```
接下来,使用 git clone 命令将 yolov5 代码克隆到本地:
```
git clone https://github.com/ultralytics/yolov5
```
最后,进入到 yolov5 文件夹中,并使用 python 运行以下命令来进行训练和测试:
```
python train.py --data data/coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 16 --epochs 100
python detect.py --weights yolov5s.pt --img 640 --conf 0.4 --source /path/to/images/or/video
```
请注意,在运行以上命令之前,需要修改coco.yaml 文件路径,配置好训练和测试的参数。
阅读全文