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

[xutao1997]

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

[ZJU-lishuang]

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

[mx2013713828]

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

[ZJU-lishuang]

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

[ruoyuchen0903]

请问如何跳过剪枝呢？

[ruoyuchen0903]

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

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