search

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

为什么稀疏训练和剪枝都要先将yolov5的一些模块给darknet，在darknet上进行操作呢？ #83

Open ztrong-forever opened 2 years ago

ztrong-forever commented 2 years ago

您好大佬！！这个问题困扰我有段时间了，查了一些资料也没有头绪，希望大佬可以解答一下，非常感谢！！

ZJU-lishuang commented 2 years ago

为了方便得到剪枝后的网络结构。 你可以想一下原版的yolov5怎么输出剪枝后的网络结构。

ztrong-forever commented 2 years ago

好的谢谢大佬！！我再研究研究！ 还有一个问题，就是这原始的yolov5的cfg文件是怎么生成的啊，这个换成自己的数据集应该还得修改修改把

ZJU-lishuang commented 2 years ago

https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects

ZJU-lishuang commented 2 years ago

原始的yolov5的cfg文件目前是手动搭建

ztrong-forever commented 2 years ago

好的谢谢大佬，我研究研究！！

ztrong-forever commented 2 years ago

image 大佬您好！！请问darknet中保存的yolov5的三个头中，mask和num参数表示什么意思啊

mx2013713828 commented 2 years ago

image 大佬您好！！请问darknet中保存的yolov5的三个头中，mask和num参数表示什么意思啊

mask=0123意味着使用前三个anchor，num=9意思是每个grid预测9个bbox。 我也在学习中，正在改cfg中。

ZJU-lishuang commented 2 years ago

num是anchor个数，mask概念来自数字图像处理，叫掩码，这里表示选择anchors中索引为0，1，2的anchor