Ubuntu18.04下通过yolov5进行训练（非anaconda版）

最近比较忙也没有更新博客，刚好今天给别人配置yolov5的，记录一下吧，希望帮到你们

一、环境

Ubuntu18.04

yolov5-6.0/python3.69

二、前期准备

2.1安装python

 由于Ubuntu18.04自带的python版本是3.6.9对应的yolov5是6.0版本的，如果想要运行高版本的则需要下载3.6以上版本的python

sudo apt-get install python3.7 #根据yolov5版本选择python版本，我的python版本是3.6.9，所以我直接下载的yolov5-6.0

python --version

2.2安装pip（选择python3的）

sudo apt install python-pip3  #(python3)
sudo apt install python pip   #(python2)

2.3下载yolov5模型

 这里我直接把u盘里的拷进来了，不会的可以去查一下怎么下载的，这里就不赘述了

2.3.1 安装包

进入到yolov5文件夹中

cd yolov5

pip3 install -U -r requirements.txt  -i https://pypi.tuna.tsinghua.edu.cn/simple

但是在安装包的过程中我出了错

ImportError: No module named skbuild

解决方法：换源最后发现豆瓣的源是可以的

pip install scikit-build -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

2.4 训练

进入yolo文件夹中，打开终端

输入指令

python3 train.py --img 640 --batch 2 --epoch 3 --data coco128.yaml --weights yolov5s.pt

最后，就可以跑通啦~

下篇更新如何训练自己的数据集！