PolyWorld: Polygonal Building Extraction with Graph Neural Networks in Satellite Images
Not on the map. Why?
Abstract (arXiv)
While most state-of-the-art instance segmentation methods produce binary segmentation masks, geographic and cartographic applications typically require precise vector polygons of extracted objects instead of rasterized output. This paper introduces PolyWorld, a neural network that directly extracts building vertices from an image and connects them correctly to create precise polygons. The model predicts the connection strength between each pair of vertices using a graph neural network and estimates the assignments by solving a differentiable optimal transport problem. Moreover, the vertex positions are optimized by minimizing a combined segmentation and polygonal angle difference loss. PolyWorld significantly outperforms the state of the art in building polygonization and achieves not only notable quantitative results, but also produces visually pleasing building polygons. Code and trained weights are publicly available at https://github.com/zorzi-s/PolyWorldPretrainedNetwork.
Key facts. Cited by 43 papers in the map (data through 14 Oct 2026). Builds on 22 of its 43 references that are in the map. First posted on arXiv 30 Nov 2021.
Builds on
22 of 43 references.
References · 22
- 2017Attention Is All You Needcited by 60,361
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2017Mask R-CNNcited by 7,277
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2019SuperGlue: Learning Feature Matching with Graph Neural Networkscited by 1,443
- 2018Path Aggregation Network for Instance Segmentationcited by 798
- 2017Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentationscited by 620
- 2018On Regularized Losses for Weakly-supervised CNN Segmentationcited by 121
- 2018Efficient Interactive Annotation of Segmentation Datasets with Polygon-RNN++cited by 120
- 2018Normalized Cut Loss for Weakly-supervised CNN Segmentationcited by 112
- 2019Fast Interactive Object Annotation with Curve-GCNcited by 95
- 2019PolyTransform: Deep Polygon Transformer for Instance Segmentationcited by 92
10 more
- 2017Annotating Object Instances with a Polygon-RNNcited by 88
- 2018TernausNet: U-Net with VGG11 Encoder Pre-Trained on ImageNet for Image Segmentationcited by 76
- 2018Topological Map Extraction from Overhead Imagescited by 58
- 2018Learning deep structured active contours end-to-endcited by 38
- 2019DARNet: Deep Active Ray Network for Building Segmentationcited by 30
- 2021MonteFloor: Extending MCTS for Reconstructing Accurate Large-Scale Floor Planscited by 26
- 2018TernausNetV2: Fully Convolutional Network for Instance Segmentationcited by 25
- 2020Machine-learned Regularization and Polygonization of Building Segmentation Maskscited by 22
- 2020Regularization of Building Boundaries in Satellite Images using Adversarial and Regularized Lossescited by 4
- 2020Quantization in Relative Gradient Angle Domain For Building Polygon Estimationcited by 3
Built on by
Citing papers per year: 2022: 5; 2023: 11; 2024: 10; 2025: 8; 2026: 9.
43 papers.
Sort:
By area:
- 2022VectorMapNet: End-to-end Vectorized HD Map Learningcited by 245
- 2023SkySense: A Multi-Modal Remote Sensing Foundation Model Towards Universal Interpretation for Earth Observation Imagerycited by 205
- 2022SatlasPretrain: A Large-Scale Dataset for Remote Sensing Image Understandingcited by 167
- 2023InstaGraM: Instance-level Graph Modeling for Vectorized HD Map Learningcited by 40
- 2023ScalableMap: Scalable Map Learning for Online Long-Range Vectorized HD Map Constructioncited by 16
- 2023GlobalMapper: Arbitrary-Shaped Urban Layout Generationcited by 15
- 20243D Building Reconstruction from Monocular Remote Sensing Images with Multi-level Supervisionscited by 13
- 2024PolygonGNN: Representation Learning for Polygonal Geometries with Heterogeneous Visibility Graphcited by 10
- 2024Pix2Poly: A Sequence Prediction Method for End-to-end Polygonal Building Footprint Extraction from Remote Sensing Imagerycited by 8
- 2022Accurate Polygonal Mapping of Buildings in Satellite Imagerycited by 7
- 2023HiT: Building Mapping with Hierarchical Transformerscited by 6
- 2024PolyR-CNN: R-CNN for end-to-end polygonal building outline extractioncited by 6
31 more
- 2024RoIPoly: Vectorized Building Outline Extraction Using Vertex and Logit Embeddingscited by 6
- 2023Patched Line Segment Learning for Vector Road Mappingcited by 5
- 2024P2PFormer: A Primitive-to-polygon Method for Regular Building Contour Extraction from Remote Sensing Imagescited by 5
- 2023Data Leakage Detection and De-duplication in Large Scale Geospatial Image Datasetscited by 4
- 2025Global Collinearity-aware Polygonizer for Polygonal Building Mapping in Remote Sensingcited by 4
- 2025HoliTracer: Holistic Vectorization of Geographic Objects from Large-Size Remote Sensing Imagerycited by 4
- 2022PolyBuilding: Polygon Transformer for End-to-End Building Extractioncited by 3
- 2024Beyond Grid Data: Exploring Graph Neural Networks for Earth Observationcited by 3
- 2025LDPoly: Latent Diffusion for Polygonal Road Outline Extraction in Large-Scale Topographic Mappingcited by 3
- 2026ACPV-Net: All-Class Polygonal Vectorization for Seamless Vector Map Generation from Aerial Imagerycited by 3
- 2023BiSVP: Building Footprint Extraction via Bidirectional Serialized Vertex Predictioncited by 1
- 20233D detection of roof sections from a single satellite image and application to LOD2-building reconstructioncited by 1
- 2024GeoFormer: A Multi-Polygon Segmentation Transformercited by 1
- 2025SAR Object Detection with Self-Supervised Pretraining and Curriculum-Aware Samplingcited by 1
- 2025The P3 Dataset: Pixels, Points and Polygons for Multimodal Building Vectorizationcited by 1
- 2025Bridging the Gap Between Sparsity and Redundancy: A Dual-Decoding Framework with Global Context for Map Inferencecited by 1
- 2025UniVector: Unified Vector Extraction via Instance-Geometry Interactioncited by 1
- 2022Primitive Graph Learning for Unified Vector Mappingcited by 0
- 2023SuperpixelGraph: Semi-automatic generation of building footprint through semantic-sensitive superpixel and neural graph networkscited by 0
- 2023Context-Enhanced Detector For Building Detection From Remote Sensing Imagescited by 0
- 2024SISP: A Benchmark Dataset for Fine-grained Ship Instance Segmentation in Panchromatic Satellite Imagescited by 0
- 2024Enhancing Polygonal Building Segmentation via Oriented Cornerscited by 0
- 2025Polygonizing Roof Segments from High-Resolution Aerial Images Using Yolov8-Based Edge Detectioncited by 0
- 2026PolyBuild: An End-to-End Method for Polygonal Building Contour Extraction from High-Resolution Remote Sensing Imagescited by 0
- 2026Vector Map as Language: Toward Unified Remote Sensing Vector Mappingcited by 0
- 2026GeoAI-based post-segmentation quality validation of building footprints via spatial feature engineeringcited by 0
- 2026SiZeUp: Fast 3D Proxy from Aerial Images via Depth Ordinal Losscited by 0
- 2026Diffusion Transformers for Roof Graph Synthesis and Reconstructioncited by 0
- 2026Topologically Consistent Agricultural Parcel Vectorization with Semantic-Guided Diffusion and Topology-Aware Polygonizationcited by 0
- 2026PolyTopoBench: A Benchmark for Complex Vector Polygon Generation from Remote Sensing Imagerycited by 0
- 2026LoDEOT: Low-Dimensional and Efficient Offset Tokens for Building Footprint Extraction from Off-Nadir Imagerycited by 0
- 2026LoDEOT: Low-Dimensional and Efficient Offset Tokens for Building Footprint Extraction from Off-Nadir Imagery5 Oct 2026
- 2026PolyTopoBench: A Benchmark for Complex Vector Polygon Generation from Remote Sensing Imagery26 Sep 2026
- 2026Topologically Consistent Agricultural Parcel Vectorization with Semantic-Guided Diffusion and Topology-Aware Polygonization7 Sep 2026
- 2026Diffusion Transformers for Roof Graph Synthesis and Reconstruction26 Aug 2026
- 2026SiZeUp: Fast 3D Proxy from Aerial Images via Depth Ordinal Loss24 Aug 2026
- 2026GeoAI-based post-segmentation quality validation of building footprints via spatial feature engineering10 Aug 2026
- 2026Vector Map as Language: Toward Unified Remote Sensing Vector Mapping9 Jun 2026
- 2026PolyBuild: An End-to-End Method for Polygonal Building Contour Extraction from High-Resolution Remote Sensing Images8 Jun 2026
- 2026ACPV-Net: All-Class Polygonal Vectorization for Seamless Vector Map Generation from Aerial Imagery17 Mar 2026
- 2025UniVector: Unified Vector Extraction via Instance-Geometry Interaction15 Oct 2025
- 2025Bridging the Gap Between Sparsity and Redundancy: A Dual-Decoding Framework with Global Context for Map Inference15 Sep 2025
- 2025HoliTracer: Holistic Vectorization of Geographic Objects from Large-Size Remote Sensing Imagery22 Jul 2025
31 more
- 2025The P3 Dataset: Pixels, Points and Polygons for Multimodal Building Vectorization21 May 2025
- 2025Global Collinearity-aware Polygonizer for Polygonal Building Mapping in Remote Sensing2 May 2025
- 2025LDPoly: Latent Diffusion for Polygonal Road Outline Extraction in Large-Scale Topographic Mapping29 Apr 2025
- 2025SAR Object Detection with Self-Supervised Pretraining and Curriculum-Aware Sampling17 Apr 2025
- 2025Polygonizing Roof Segments from High-Resolution Aerial Images Using Yolov8-Based Edge Detection12 Mar 2025
- 2024Pix2Poly: A Sequence Prediction Method for End-to-end Polygonal Building Footprint Extraction from Remote Sensing Imagery10 Dec 2024
- 2024GeoFormer: A Multi-Polygon Segmentation Transformer25 Nov 2024
- 2024Beyond Grid Data: Exploring Graph Neural Networks for Earth Observation5 Nov 2024
- 2024RoIPoly: Vectorized Building Outline Extraction Using Vertex and Logit Embeddings20 Jul 2024
- 2024PolyR-CNN: R-CNN for end-to-end polygonal building outline extraction20 Jul 2024
- 2024Enhancing Polygonal Building Segmentation via Oriented Corners17 Jul 2024
- 2024PolygonGNN: Representation Learning for Polygonal Geometries with Heterogeneous Visibility Graph30 Jun 2024
- 2024P2PFormer: A Primitive-to-polygon Method for Regular Building Contour Extraction from Remote Sensing Images5 Jun 2024
- 20243D Building Reconstruction from Monocular Remote Sensing Images with Multi-level Supervisions7 Apr 2024
- 2024SISP: A Benchmark Dataset for Fine-grained Ship Instance Segmentation in Panchromatic Satellite Images6 Feb 2024
- 2023SkySense: A Multi-Modal Remote Sensing Foundation Model Towards Universal Interpretation for Earth Observation Imagery15 Dec 2023
- 2023ScalableMap: Scalable Map Learning for Online Long-Range Vectorized HD Map Construction20 Oct 2023
- 2023Context-Enhanced Detector For Building Detection From Remote Sensing Images11 Oct 2023
- 2023HiT: Building Mapping with Hierarchical Transformers18 Sep 2023
- 2023Patched Line Segment Learning for Vector Road Mapping6 Sep 2023
- 2023GlobalMapper: Arbitrary-Shaped Urban Layout Generation19 Jul 2023
- 20233D detection of roof sections from a single satellite image and application to LOD2-building reconstruction11 Jul 2023
- 2023SuperpixelGraph: Semi-automatic generation of building footprint through semantic-sensitive superpixel and neural graph networks12 Apr 2023
- 2023Data Leakage Detection and De-duplication in Large Scale Geospatial Image Datasets5 Apr 2023
- 2023BiSVP: Building Footprint Extraction via Bidirectional Serialized Vertex Prediction1 Mar 2023
- 2023InstaGraM: Instance-level Graph Modeling for Vectorized HD Map Learning10 Jan 2023
- 2022SatlasPretrain: A Large-Scale Dataset for Remote Sensing Image Understanding28 Nov 2022
- 2022PolyBuilding: Polygon Transformer for End-to-End Building Extraction3 Nov 2022
- 2022Accurate Polygonal Mapping of Buildings in Satellite Imagery1 Aug 2022
- 2022Primitive Graph Learning for Unified Vector Mapping28 Jun 2022
- 2022VectorMapNet: End-to-end Vectorized HD Map Learning17 Jun 2022
- 2025LDPoly: Latent Diffusion for Polygonal Road Outline Extraction in Large-Scale Topographic Mappingcited by 3
- 2026Diffusion Transformers for Roof Graph Synthesis and Reconstructioncited by 0
- 2026Topologically Consistent Agricultural Parcel Vectorization with Semantic-Guided Diffusion and Topology-Aware Polygonizationcited by 0
- 2026Topologically Consistent Agricultural Parcel Vectorization with Semantic-Guided Diffusion and Topology-Aware Polygonization7 Sep 2026
- 2026Diffusion Transformers for Roof Graph Synthesis and Reconstruction26 Aug 2026
- 2025LDPoly: Latent Diffusion for Polygonal Road Outline Extraction in Large-Scale Topographic Mapping29 Apr 2025
- 2022VectorMapNet: End-to-end Vectorized HD Map Learningcited by 245
- 2023SkySense: A Multi-Modal Remote Sensing Foundation Model Towards Universal Interpretation for Earth Observation Imagerycited by 205
- 2022SatlasPretrain: A Large-Scale Dataset for Remote Sensing Image Understandingcited by 167
- 2023InstaGraM: Instance-level Graph Modeling for Vectorized HD Map Learningcited by 40
- 2023ScalableMap: Scalable Map Learning for Online Long-Range Vectorized HD Map Constructioncited by 16
- 2023GlobalMapper: Arbitrary-Shaped Urban Layout Generationcited by 15
- 20243D Building Reconstruction from Monocular Remote Sensing Images with Multi-level Supervisionscited by 13
- 2024PolygonGNN: Representation Learning for Polygonal Geometries with Heterogeneous Visibility Graphcited by 10
- 2024Pix2Poly: A Sequence Prediction Method for End-to-end Polygonal Building Footprint Extraction from Remote Sensing Imagerycited by 8
- 2022Accurate Polygonal Mapping of Buildings in Satellite Imagerycited by 7
- 2023HiT: Building Mapping with Hierarchical Transformerscited by 6
- 2024PolyR-CNN: R-CNN for end-to-end polygonal building outline extractioncited by 6
12 most cited of 40 All 40 (JSON)
- 2026LoDEOT: Low-Dimensional and Efficient Offset Tokens for Building Footprint Extraction from Off-Nadir Imagery5 Oct 2026
- 2026PolyTopoBench: A Benchmark for Complex Vector Polygon Generation from Remote Sensing Imagery26 Sep 2026
- 2026SiZeUp: Fast 3D Proxy from Aerial Images via Depth Ordinal Loss24 Aug 2026
- 2026GeoAI-based post-segmentation quality validation of building footprints via spatial feature engineering10 Aug 2026
- 2026Vector Map as Language: Toward Unified Remote Sensing Vector Mapping9 Jun 2026
- 2026PolyBuild: An End-to-End Method for Polygonal Building Contour Extraction from High-Resolution Remote Sensing Images8 Jun 2026
- 2026ACPV-Net: All-Class Polygonal Vectorization for Seamless Vector Map Generation from Aerial Imagery17 Mar 2026
- 2025UniVector: Unified Vector Extraction via Instance-Geometry Interaction15 Oct 2025
- 2025Bridging the Gap Between Sparsity and Redundancy: A Dual-Decoding Framework with Global Context for Map Inference15 Sep 2025
- 2025HoliTracer: Holistic Vectorization of Geographic Objects from Large-Size Remote Sensing Imagery22 Jul 2025
- 2025The P3 Dataset: Pixels, Points and Polygons for Multimodal Building Vectorization21 May 2025
- 2025Global Collinearity-aware Polygonizer for Polygonal Building Mapping in Remote Sensing2 May 2025
12 newest of 40 All 40 (JSON)