Python部署YOLOV5-cpu环境快速体验目标检测
Python部署YOLOV5-cpu环境快速体验目标检测
目录大纲:
安装torch:
点我访问torch官网
到官网复制命令安装torch-cpu版本:
pip3 install torch torchvision torchaudio
注释:如果pip3安装失败可以尝试直接:
pip install torch torchvision torchaudio
接下来进入到yolov5根目录找到requirements.txt
cd /d xxxx/xxx/xx
修改里面部分内容,将# torch>=1.7.0 和torchvision>=0.8.1注释掉,因为前面我们已经单独安装过torch了
输入命令安装YOLOV5所有支持的库:
pip install -r requirements.txt
安装完成后的示例图:
安装git模块:
因为运行代码会提示报错,要求安装git软件才行所以我们也跟着安装
先安装git软件(win电脑):
点我访问git官网进行下载
注释安装git记得要重启电脑,一定重启他才会生效
重启电脑后,接着安装git的python模块:
pip install gitpython
运行代码测试成果:
改进train.py代码中–batch-size为1,workers为1
接着运行train.py:
运行代码的时候他会自动下载coco128数据集,这里大家记得上梯子否则会保持下载失败
运行成功后的示例图: