Project for the class on Image Processing for Earth Observation - EPFL, 2023
In this project, we replicate Deforestation Alerts from the Global Forest Watch Integrated Alert System with various deep learning methods with data covering the Amazon Rainforest. In doing so, we explore the impact of working with an additional NIR channel besides the optical R-G-B bands. Our best final model leverages the MSNet architecture with all four input bands (R-G-B-NIR) and reaches an F1-Score of 0.9107 and an Accuracy of 0.9164.
To code is structured as a package. To use this repository, you should first create a virtual enivronment to install the required dependencies as well as the repository itself. For instance with conda, run
conda create --name project_deforestation python=3.9
conda activate project_deforestation
pip install -r requirements.txt
pip install -e .Important: Adjust the PATH_TO_DATA variable in the config/config.py file to point to the location of the data.
To run an example inference for one image, you can use the inference.ipynb notebook located in the root directory.
To do so, you must download the final model weights via this Google Drive Link and place the weights into the project's root directory.
Two examples predictions and their respective ground truth labels are shown below:
The dataset is made available through the Norway's International Climate and Forests Initiative Satellite Data Program. The present work used a curated subset for training. The labels are sourced from the Integrated Alert System from the Global Forest Watch initiative (see a short description here).
Two examples of the data can be seen below:
Source Files:
- model: Package directory for storing model implementations.
- utils: Package directory for utility functions and helper scripts.
Notebooks:
- A. Data Exploration
A1_Explore_Data.ipynb: Notebook for data exploration and testing the data loaders.
- B. Model Training & Evaluation Notebooks
B0_MLE_benchmark.ipynb: Notebook for creating and evaluating the benchmark model based on Multivariate Gaussian Maximum Likelhood Estimation.B1_Deeplabv3plus.ipynb: Notebook for training and running experiments with the DeepLabv3+ Model.B2_MSNet.ipynb: Notebook for training and running experiments with the MSNet Model.B3_UNet.ipynb: Notebook for training and running experiments with the UNet Model.
Setup & Configuration:
config/config.py: Configuration file to change path to data.requirements.txt: File specifying the Python dependencies.pyproject.toml: Project build file.