Downloads · 30 days
141
7% of all-time downloads
ibm-granite/granite-geospatial-uki-flooddetection
granite-geospatial-uki-flooddetection is a image segmentation model from ibm-granite. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. It is set up for terratorch. The card lists the license as apache-2.0.
<b<iTry it on Colab<<</i</b (Please select T4 runtime)
Downloads · 30 days
141
7% of all-time downloads
All-time downloads
2.1K
Public
Repo size
2.3 GB
Likes
21
Public
Click a slice to open those files.
.ckpt1.1 GB · 100%
From the Hugging Face model README
<b><i>>>Try it on Colab<<</i></b> (Please select T4 runtime)
This is a fine-tuned geospatial foundation model for detecting flood and surface water in the United Kingdom and Ireland using multispectral and synthetic aperture radar (SAR) satellite imagery.
The model predicts areas of water from Sentinel-2 and Sentinel-1 SAR imagery and was trained on flood events from the United Kingdom and Ireland, recorded by the Copernicus Emergency Management service (CEMS). Please see the model description and training Data below for more details. The model was fine-tuned from the granite-geospatial-uki model, which has also been released.
You can use granite-geospatial-flood-detection-uki to run inference and detect flood events. Our experiments have shown that while granite-geospatial-flood-detection-uki works best in the UK and Ireland, the base model granite-geospatial-uki can also be successfully fine-tuned for flood detection in other locations. Please see the granite-geospatial-uki model card for details.
<figure> <img src='flood_image.jpg' alt='missing' /> <figcaption>Flooding on Hanley Road by Philip Halling, CC BY-SA 2.0 <https://creativecommons.org/licenses/by-sa/2.0>, via Wikimedia Commons </figcaption> </figure>Inference notebook, for running the granite-geospatial-flood-detection-uki model.
Fine-tuning notebooks for fine tuning on other locations using the base model granite-geospatial-uki
For fine-tuning, we replaced the decoder used in the granite-geospatial-uki model with a fully convolutional decoder and performed a sematic segmentation task with two classes, either no water or surface water. Some flood events were obscured by clouds in the Sentinel-2 data, and we found that including a binary cloud mask in the fine-tuning data as an additional band significantly improved performance.
The bands that were used in the fine-tuned model are the following:
The UK and Ireland data consisted of 69, 512x512 pixel images, from 6 separate flood events. Of the 69 images, 50 were used for training, 10 for validation and 9 for testing. Both cloudy and cloud-free images were included in the dataset.
Helen Tamura-Wicks, Andrew Taylor, Chris Dearden, Geoffrey Dawson, Paolo Fraccaro & Anne Jones.
This work was supported by the Hartree National Centre for Digital Innovation, a collaboration between STFC and IBM.
All content in this repository including code has been provided by IBM under the associated open source software license and IBM is under no obligation to provide enhancements, updates, or support. IBM developers produced this code as an open source project (not as an IBM product), and IBM makes no assertions as to the level of quality nor security, and will not be maintaining this code going forward.