1. 下载项目文件
从百度网盘下载并解压
网盘地址,文末有链接
:包括项目完整源代码、数据集、原理的课件说明等。
- 解压源码: yolov5-6.1-slimming.zip
项目没有从yolov5 github上直接克隆项目文件,而是从百度网盘上下载项目文件并解压,因为yolov5原始的代码是没有带网络剪枝的,网盘上的代码是加上了网络剪枝的完整代码
,项目是基于yolov5 6.1版本上修改的。
2. 安装所需库
使用清华镜像源,在yolov5-6.1-slimming路径下执行:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
注意:simple 不能少, 是 https 而不是 http
3 下载预训练权重文件
yolov5的github 链接:https://github.com/ultra