A Rubust Method for Roof Extraction and Height Estimation

Abstract

In this article, we introduce the solution we used in the 2023 IEEE GRSS Data Fusion Contest Track 2. The task demands a roof type classification method and a building height estimation method. For roof type classification, our experiments are based on Swin transformer, combined with DoubleHead module and RFLA strategy, which can effectively improve model’s performance on small objects. For building height estimation, our experiments are based on SegFormer. Our experiments use a part of train set as validation set.