YOLOv5_OBB_KLD icon indicating copy to clipboard operation
YOLOv5_OBB_KLD copied to clipboard

大佬有论文吗

Open Yeawha opened this issue 1 year ago • 4 comments

大佬有没有出论文拜读一下

Yeawha avatar Mar 03 '23 10:03 Yeawha

没有,这个改动太小了。如果能把样本匹配方式改成适合高斯分布的方式的话,那样还不错。

lx-cly avatar Mar 05 '23 08:03 lx-cly

大佬,你说的样本匹配方式是指哪一块呢?

Freeman-Cai avatar Aug 17 '23 05:08 Freeman-Cai

大佬,你说的样本匹配方式是指哪一块呢?

yolov5的样本匹配代码实现在loss.py->Class ComputeLoss->def build_targets,选取距离GT中心点最近的3个cell的anchor,计算anchor与GT的高宽比例,比较阈值实现正负样本分配。现在bbox表示改成高斯分布,只修改损失函数的计算,改动不多,提升也并不是很大。 或许这篇论文可以帮助到你。General Gaussian Heatmap Label Assignment for Arbitrary-Oriented Object Detection

lx-cly avatar Aug 17 '23 06:08 lx-cly

大佬,你说的样本匹配方式是指哪一块呢?

yolov5的样本匹配代码实现在loss.py->Class ComputeLoss->def build_targets,选取距离GT中心点最近的3个cell的anchor,计算anchor与GT的高宽比例,比较阈值实现正负样本分配。现在bbox表示改成高斯分布,只修改损失函数的计算,改动不多,提升也并不是很大。 或许这篇论文可以帮助到你。General Gaussian Heatmap Label Assignment for Arbitrary-Oriented Object Detection

感谢大佬,我去看看这篇论文,看有没有什么想法

Freeman-Cai avatar Aug 17 '23 06:08 Freeman-Cai