yolov5剪枝实战3: yolov5-slimming项目运行演示

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

你可能感兴趣的:(模型轻量化,剪枝,python,人工智能)