YOLOv5目标检测之网络剪枝实战
课时介绍
实战篇包括:PyTorch环境安装、YOLOv5项目安装、准备自己的数据集、修改配置文件、正常训练自己的数据集、稀疏化训练、网络模型剪枝、剪枝后的网络模型微调。
课程介绍
PyTorch版的YOLOv5是一个非常流行的基于深度学习的目标检测器。本课程使用Network Slimming剪枝方法对YOLOv5进行网络剪枝,使其更加轻量。
Network Slimming是一种经典的实用模型压缩方法,可实现高效的通道级别的结构化剪枝。该方法通过利用BN(Batch Normalization)层中缩放因子较小的值来剪裁的相应通道,达到精简网络的目的。步骤包括:增加稀疏正则化的训练、网络模型剪枝、剪枝后的网络模型微调。
本课程在YOLOv5 v6.1版本代码的基础上增加Network Slimming剪枝方法,在Windows和Ubuntu系统上演示针对自己的数据集训练和进行网络剪枝过程,并讲解原代码针对剪枝的修改部分。
本课程分为原理篇、实战篇、代码讲解篇。
- 原理篇包括:网络剪枝的基础知识、Network Slimming剪枝原理。
- 实战篇包括:PyTorch环境安装、YOLOv5项目安装、准备自己的数据集、修改配置文件、正常训练自己的数据集、稀疏化训练、网络模型剪枝、剪枝后的网络模型微调。
- 代码讲解篇包括:剪枝代码修改说明和具体修改的代码讲解。
推荐课程
信息系统项目管理师自考笔记
李明 · 522人在学
python从0到1:期货量化交易系统(CTP实战,高频及合成K线数据
王先生 · 20718人在学
手把手搭建Java超市管理系统【附源码】(毕设)
汤小洋 · 4074人在学
Java毕设springboot外卖点餐系统 毕业设计毕设源码 使用教
黄菊华 · 734人在学
基于SSM酒店管理系统(毕设)
小尼老师 · 790人在学
java项目实战之购物商城(java毕业设计)
Long · 5092人在学
手把手搭建Java求职招聘系统【附源码】(毕设)
汤小洋 · 1463人在学
Python Django 深度学习 小程序
钟翔 · 2245人在学
城管局门前三包管理系统+微信小程序(vue+springboot)
赖国荣 · 508人在学
Vue+Uni-app(uniapp)入门与实战+赠送仿美团点餐小程序
李杰 · 3855人在学