YOLOv5_OBB_KLD
YOLOv5_OBB_KLD copied to clipboard
大佬有论文吗
大佬有没有出论文拜读一下
没有,这个改动太小了。如果能把样本匹配方式改成适合高斯分布的方式的话,那样还不错。
大佬,你说的样本匹配方式是指哪一块呢?
大佬,你说的样本匹配方式是指哪一块呢?
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
大佬,你说的样本匹配方式是指哪一块呢?
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
感谢大佬,我去看看这篇论文,看有没有什么想法