May, 2022

使用热图加权损失的轻量级人体姿态估计

TL;DR本文提出了一种减轻资源负荷、加速推理速度的方法,是通过在 SimpleBaseline 的反卷积头网络中引入注意机制来利用原始、跨层次和内层次信息以提高精度,并采用称为 heatmap 加权损失的新型损失函数,生成热图上每个像素的权重,使模型更加注重关键点,实验证明我们的方法在性能和资源和推理速度之间实现了平衡,具有不错的适用性。