yolov5_prune
yolov5_prune copied to clipboard
为什么稀疏训练和剪枝都要先将yolov5的一些模块给darknet,在darknet上进行操作呢?
您好大佬!!这个问题困扰我有段时间了,查了一些资料也没有头绪,希望大佬可以解答一下,非常感谢!!
为了方便得到剪枝后的网络结构。 你可以想一下原版的yolov5怎么输出剪枝后的网络结构。
好的谢谢大佬!!我再研究研究! 还有一个问题,就是这原始的yolov5的cfg文件是怎么生成的啊,这个换成自己的数据集应该还得修改修改把
https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects
原始的yolov5的cfg文件目前是手动搭建
好的谢谢大佬,我研究研究!!
大佬您好!!请问darknet中保存的yolov5的三个头中,mask和num参数表示什么意思啊
大佬您好!!请问darknet中保存的yolov5的三个头中,mask和num参数表示什么意思啊
mask=0123意味着使用前三个anchor,num=9意思是每个grid预测9个bbox。 我也在学习中,正在改cfg中。
num是anchor个数,mask概念来自数字图像处理,叫掩码,这里表示选择anchors中索引为0,1,2的anchor