cc-ai / kdb

AI can help Visualizing the Impacts of Climate Change. This is an open forum to share our work
https://mila.quebec/en/ai-society/visualizing-climate-change/
Apache License 2.0
19 stars 3 forks source link

Improve flood segmentation model #109

Open melisandeteng opened 4 years ago

melisandeteng commented 4 years ago

Improve the current flood segmentation model , see #62

We report below a comparison of a first series of flood segmentation models using mIoU on our hand labeled dataset of 209 images from Keymakr.

DeepLab v2 with a ResNet-101 backbone (current model):

DeepLab v2 with a ResNet-101 backbone + CRF postprocessing: mIoU ~ 80 % Extremely slow (> 10s / image)

Sazara et. al - pretrained VGG-16 + CRF postprocessing: fine tuned on 202 hand-labeled images mIoU ~ 77% slow : 7s/ image (and images had to be resized to have a max size of 600*600 pixels)

melisandeteng commented 4 years ago

Note that the masks for the flooded images will only need to be computed once. Next steps include :

vict0rsch commented 4 years ago

How's that going @melisandeteng ?