mapbox / robosat

Semantic segmentation on aerial and satellite imagery. Extracts features such as: buildings, parking lots, roads, water, clouds
MIT License
2.02k stars 382 forks source link

Feature pyramid attention (FPA) modules #167

Open daniel-j-h opened 5 years ago

daniel-j-h commented 5 years ago

We should look into the feature pyramid attention (FPA) module for pixel-precise attention for segmentation features extracted from our resnet encoder.

Pyramid Attention Network for Semantic Segmentation https://arxiv.org/abs/1805.10180

fpa-0

from https://arxiv.org/abs/1805.10180 Figure 2

fpa-1

from https://arxiv.org/abs/1805.10180 Figure 3

The GAU modules could be interesting to try out instead of our simple decoder modules, too.