Polygonal Building Segmentation by Frame Field Learning
Polygonization-by-Frame-Field-Learning
This repository contains the code for our fast polygonal building extraction from overhead images pipeline.
We add a frame field output to an image segmentation neural network to improve segmentation quality and provide structural information for the subsequent polygonization step.
Figure 1: Close-up of our additional frame field output on a test image.
Figure 2: Given an overhead image, the model outputs an edge mask, an interior mask,
and a frame field for buildings. The total loss includes terms that align the masks and
frame field to ground truth data as well as regularizers to enforce smoothness of the
frame field and consistency between the outputs.
Figure 3: Given classification maps and a frame field as input, we optimize skeleton polylines to
align