arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2302.01585cs.CV

SegForestNet:基于空间划分的航空图像分割

SegForestNet: Spatial-Partitioning-Based Aerial Image Segmentation

Leibniz Universität Hannover(莱布尼茨汉诺威大学)

Daniel Gritzner, Jörn Ostermann

AI总结:

本文改进预测BSP树的SegForestNet,通过新解码器、可微渲染器和损失函数实现多边形式航空图像分割,并揭示训练优化对架构优势的影响。

摘要:

航空图像分割是自动制图或森林砍伐追踪等应用的基础。在这些应用中常用的真正射影像里,许多物体和区域都可以用多边形很好地近似。然而,这一事实很少被最先进的语义分割模型利用。相反,大多数模型允许任意区域形状,从而在预测中引入了不必要的自由度。因此,我们提出了对本团队深度学习模型的一种改进,该模型预测二元空间划分树,这是一种高效的多边形表示。改进包括一种新的特征解码器架构和一种新的可微BSP树渲染器,二者都可避免梯度消失。此外,我们设计了一种新的损失函数,专门用于改善由预测树定义的空间划分。进一步地,扩展后的模型可以一次预测多棵树,因此能够预测特定类别的分割。作为另一项贡献,我们研究了与优化训练过程相比,非最优训练过程的影响。尽管为航空图像优化的模型架构(如PFNet或我们自己的模型)在非最优条件下表现出优势,但这种优势在最优训练条件下会消失。尽管观察到这一点,我们的模型对小型矩形物体(例如汽车)仍能作出更好的预测。

英文摘要:

Aerial image segmentation is the basis for applications such as automatically creating maps or tracking deforestation. In true orthophotos, which are often used in these applications, many objects and regions can be approximated well by polygons. However, this fact is rarely exploited by state-of-the-art semantic segmentation models. Instead, most models allow unnecessary degrees of freedom in their predictions by allowing arbitrary region shapes. We therefore present a refinement of our deep learning model which predicts binary space partitioning trees, an efficient polygon representation. The refinements include a new feature decoder architecture and a new differentiable BSP tree renderer which both avoid vanishing gradients. Additionally, we designed a novel loss function specifically designed to improve the spatial partitioning defined by the predicted trees. Furthermore, our expanded model can predict multiple trees at once and thus can predict class-specific segmentations. As an additional contribution, we investigate the impact of a non-optimal training process in comparison to an optimized training process. While model architectures optimized for aerial images, such as PFNet or our own model, show an advantage under non-optimal conditions, this advantage disappears under optimal training conditions. Despite this observation, our model still makes better predictions for small rectangular objects, e.g., cars.

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

↑