网络结构改了该怎么修改您的代码进行剪枝呢

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5

Apache License 2.0

556 stars 136 forks source link

Open xutao1997 opened 2 years ago

xutao1997 commented 2 years ago

我在网络结构增加了一个检测层该怎么修改您的代码进行剪枝呢

ZJU-lishuang commented 2 years ago

在网络框架cfg和yaml中添加相应的检测层，yaml用于训练，cfg用于剪枝和生成剪枝后的网络结构，model_transfer.py 文件中模型转换添加相应的检测层。

mx2013713828 commented 2 years ago

顺便问一下，使用添加了transfromer版本的yolov5可以使用这个剪枝吗？我尝试了官方提供的transformer版本速度精度提升效果都不错，想在这基础上剪枝。 anyway，今晚我准备实验一下。

ZJU-lishuang commented 2 years ago

对于不支持的层可以跳过剪枝，剪枝的基础结构是conv+bn，根据bn参数减少conv通道数

ruoyuchen0903 commented 1 year ago

请问如何跳过剪枝呢？

ruoyuchen0903 commented 1 year ago

顺便问一下，使用添加了transfromer版本的yolov5可以使用这个剪枝吗？我尝试了官方提供的transformer版本速度精度提升效果都不错，想在这基础上剪枝。 anyway，今晚我准备实验一下。

您好，请问这个实验您做了吗？如果对添加transformer的yolov5剪枝呢？如果是跳过添加层，如果修改代码呢？