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.csdn.net/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.csdn.net/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"