一种基于热力图的高密度豆粒计数及中心点定位方法和系统
摘要:
一种基于热力图的高密度豆粒计数及中心点定位方法,用高斯函数生成高斯核模板,结合已标记的豆粒中心点位置,生成用于豆粒计数的真值热力图;采用基于空洞卷积的CSRNet作为密度图估计模块,将原始图像与真值热力图输入到模型中计算得到与原始图像同大小的热力图,通过对比预测热力图与真值热力图的L2损失进行参数的学习,实现高质量的热力图估计。对于待测试图像,使用CSRNet预测热力图,再通过判断局部最大位置点,从热力图中获取得到所有中心点的位置坐标,并通过局部中心点热力图的值取整获得豆粒数。还包括一种基于热力图的高密度豆粒计数及中心点定位系统。本发明可提高豆粒计数模型在高密度、遮挡严重场景下的计数准确性。
0/0