中文

PMODE:基于原型掩码的目标尺寸估计

计算机视觉与模式识别 2022-12-29 v1

摘要

神经网络能否在真实场景中估计物体的尺寸?本文中,我们提出一种方法和深度学习架构,利用单目相机估计视频中感兴趣四边形物体的尺寸。所提技术不使用相机标定或手工几何特征;然而,特征是在训练过程中借助分割神经网络的系数学习的。我们采用了基于实时实例分割的深层神经网络,其骨干为 ResNet50,给出物体的原型掩码从而提供感兴趣区域以回归其尺寸。实例分割网络经训练仅关注最近的感兴趣物体。回归通过 MLP 头执行,其仅关注边界框检测头的掩码系数与原型分割掩码。我们使用三种不同随机相机训练系统,在尺寸估计测试数据集上取得了 22% 的 MAPE。

关键词

引用

@article{arxiv.2212.13281,
  title  = {PMODE: Prototypical Mask based Object Dimension Estimation},
  author = {Thariq Khalid and Mohammed Yahya Hakami and Riad Souissi},
  journal= {arXiv preprint arXiv:2212.13281},
  year   = {2022}
}

备注

10 pages