yolov5_prune icon indicating copy to clipboard operation
yolov5_prune copied to clipboard

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

Open ztrong-forever opened this issue 3 years ago • 8 comments

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

ztrong-forever avatar Jan 18 '22 15:01 ztrong-forever

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

ZJU-lishuang avatar Jan 19 '22 01:01 ZJU-lishuang

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

ztrong-forever avatar Jan 19 '22 01:01 ztrong-forever

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

ZJU-lishuang avatar Jan 19 '22 01:01 ZJU-lishuang

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

ZJU-lishuang avatar Jan 19 '22 01:01 ZJU-lishuang

好的谢谢大佬,我研究研究!!

ztrong-forever avatar Jan 19 '22 01:01 ztrong-forever

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

ztrong-forever avatar Jan 19 '22 07:01 ztrong-forever

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

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

mx2013713828 avatar Jan 29 '22 06:01 mx2013713828

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

ZJU-lishuang avatar Jan 29 '22 07:01 ZJU-lishuang