arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.09770cs.CV
收藏

MobileStereoNet:面向立体匹配的轻量级深度网络

MobileStereoNet: Towards Lightweight Deep Networks for Stereo Matching

University of Tuebingen(图宾根大学);WSI Institute for Computer Science(WSI计算机科学研究所)

Faranak Shamsafar, Samuel Woerz, Rafia Rahim, Andreas Zell

摘要:

最近的立体匹配方法使用深度模型不断提高精度。然而,这种增益是在计算成本大幅增加的情况下获得的,使得网络甚至可能不适合中等GPU。当模型需要部署在资源有限的设备上时,此问题会引起问题。为此,我们提出了两个立体视觉的光模型,降低了复杂性,而不牺牲精度。根据成本体积的维度,我们设计了一个2D和一个3D模型,分别从2D和3D卷积构建编码器-解码器。为此,我们利用2D MobileNet块,并将其扩展到3D立体视觉应用。此外,还提出了一个新的成本体积,以提高2D模型的准确性,使其性能接近3D网络。实验表明,所提出的2D/3D网络有效地减少了计算开销(2D和3D模型中的参数/操作分别减少了27%/95%和72%/38%),同时保持了准确性。我们的代码可在https://github.com/cogsys-tuebingen/mobilestereonet上获得。

英文摘要:

Recent methods in stereo matching have continuously improved the accuracy using deep models. This gain, however, is attained with a high increase in computation cost, such that the network may not fit even on a moderate GPU. This issue raises problems when the model needs to be deployed on resource-limited devices. For this, we propose two light models for stereo vision with reduced complexity and without sacrificing accuracy. Depending on the dimension of cost volume, we design a 2D and a 3D model with encoder-decoders built from 2D and 3D convolutions, respectively. To this end, we leverage 2D MobileNet blocks and extend them to 3D for stereo vision application. Besides, a new cost volume is proposed to boost the accuracy of the 2D model, making it performing close to 3D networks. Experiments show that the proposed 2D/3D networks effectively reduce the computational expense (27%/95% and 72%/38% fewer parameters/operations in 2D and 3D models, respectively) while upholding the accuracy. Our code is available at https://github.com/cogsys-tuebingen/mobilestereonet.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑