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
18 stars 3 forks source link

Classifying Flooded image vs non-flooded Image #92

Closed gcosne closed 4 years ago

gcosne commented 5 years ago

Developing a good classifier that would recognize a Flooded image vs a non-flooded one could be useful for many applications.

We intend to use the classifier as a supplementary information to guide the image-to-image translation process at training time.

Architecture proposed: As we would like a light model that generalizes well with a small amount of training examples, a good baseline would be to fine-tune a small resnet architecture trained on image net as presented in the following tutorial pytorch-tutorial.

gcosne commented 5 years ago

First version of the flood classifier: cc-ai/floods-gans/flood-classifier