lishuang
lishuang
这些文件是剪枝自动生成的文件,你跑一遍程序就会有的
类别数检查一下
要看网络层是否支持。也就是代码是否能解析cfg文件。不支持可以在代码里添加对不支持的层的支持。
refer to [yolov3-channel-and-layer-pruning](https://github.com/tanluren/yolov3-channel-and-layer-pruning)
https://github.com/ZJU-lishuang/yolov5_prune/blob/v6/shortcut_prune_yolov5s.py#L67 这里注释看一下。 报错是没有发现文件,是不是没权限创建文件
cfg文件就是网络模型的结构
目前整理复现了该步骤,未出现该问题。请更新最新版本。
为了方便得到剪枝后的网络结构。 你可以想一下原版的yolov5怎么输出剪枝后的网络结构。
https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects