linux :YOLO5配置(ubuntu 20.04 下安装运行yolov5)

ubuntu 20.04 下安装运行yolov5

新建文件夹，安装python

首先建立一个文件夹随便命名就好(保证自己以后可以方便找到)，然后保证自己的python的版本要大于3：下载python3.7
ubuntu自带python 3.8.5

sudo apt-get install python3.7

安装 python-pip：

sudo apt install python-pip

安装pytorch和 torchvision

方法1：

(base) dragon@dragon-virtual-machine:~/Desktop$ conda config --set auto_activate_base false
(base) dragon@dragon-virtual-machine:~/Desktop$ conda activate yolov5
(yolov5) dragon@dragon-virtual-machine:~/Desktop$ conda install pytorch torchvision cudatoolkit=10.2

方法2：

本教程中使用的是pytorch1.7.1和 torchvision 0.8.2

命令：
pip install torchvision
在anaconda下：

conda install torchvision -c pytorch
下载地址：

https://download.pytorch.org/whl/torch_stable.html

克隆YOLOv5Git code

在该文件夹下打开终端，将yolov5下载到该文件下

git clone https://github.com/ultralytics/yolov5.git

配置YOLO环境

然后进入yolov5文件夹下

cd yolov5

配置yolov5环境：

pip3 install -U -r requirements.txt 

E1:
WARNING: The scripts f2py, f2py3 and f2py3.8 are installed in ‘/home/dragon/.local/bin’ which is not on PATH.
Consider adding this directory to PATH or, if you prefer to suppress this warning, use --no-warn-script-location.
意思是你没有将环境变量添加到路径，如果忽略这个警告，后面会发现无法导入包，

解决方案 环境变量配置：运行 gedit ~/.bashrc
在终端中输入：

gedit ~/.bashrc

然后把export PATH=/home/dragon/.local/bin/:$PATH放到最后
然后更新文件。(dragon 是用户名需要根据本地账户进行修改)

source ~/.bashrc  

可能会出错No module named “skbuild”:

pip3 install scikit-build

参考链接：https://blog.youkuaiyun.com/yiminghd2861/article/details/98854882

安装yolov5:

pip3 install --upgrade pillow -i https://pypi.tuna.tsinghua.edu.cn/simple

下载。yoloV5.pt
git clone https://gitee.com/seanxyz/yolov5_demo.git

这是下载安装好的文件夹：
————————————————

原文链接：https://blog.youkuaiyun.com/weixin_51326570/article/details/113803695

训练

进入 yoloV5文件夹，激活虚拟环境运行如下指令
python3 train.py --img 640 --batch 16 --epochs 5 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ‘’
根据提示安装需要的libs;

测试时使用的是虚拟机，没用GPU，开始训练一会就停了，

看到下面结果说明训练数据可以开始了：

测试

python3 detect.py --source ./inference/images/ --weights runs/exp16/weights/best.pt --conf 0.4 --device cpu --img-size 320

python3 detect.py --source "data/images/bus.jpg" --weights="weights/yolov5s.pt"