摘要

在侧面脊柱关键点检测时,由于受到器官遮挡的影响,以往的热图回归方法难以区分不同椎骨上的关键点,容易出现关键点与对应椎骨的匹配错误。为了解决这个问题,本文提出了一个新的侧面脊柱关键点检测方法,该方法同时预测关键点热图和关键点匹配线索(椎骨中心热图和关键点offset),利用匹配线索建立关键点与对应椎骨的匹配关系。为了提升匹配效果,本文提出几何感知特征增强模块(Geometry-aware feature aggregator,GFA),该模块通过提取关键点特征增强椎骨中心的特征表达。本文还利用加权损失函数缓解关键点热图和椎骨中心热图中正负样本比例失衡问题。实验结果表明,本文方法的平均检测误差为8.84,相较于性能第二的方法精度提升36%。