This project was for the ESA 2019 Summer of Code
Cloud detection with machine learning using the RGBN bands
This algorithm started off as just applying band thresholding and other filters and then progressed into using algorithms, such as K-Means Clustering and Quickshift to cluster the image, to try and detect and mask clouds. A big issue that I’ve come across is trying to detect haze in an image. V1 also contains cloud shadow masking that is based upon the generated cloud mask. A problem with this is if there’s a cloud shadow in the image but there’s no cloud, it won’t get detected. Otherwise, it's pretty decent at detecting cloud shadows.
V1 does contain a bit of machine learning, as it is just an initial attempt at solving this issue and so that I can learn the basics, but more advanced machine learning (CNN’s) is definitely going to be explored in V2. The included model and training data is pretty limited (only 80 or so samples) so the SVM's performance at classifying clusters is not the greatest. However, with more training data and more inputs, such as the GLCM outputs, I believe it would be able to perform better.
Another area to look into would be going back to using the K-Means Clustering instead of the Gaussian Mixture Model. I chose to use the GMM because it generally clustered pixels more accurately, however it is much, much slower than the K-Means algorithm in scikit-learn
because (among other things like covariance_type
and num_clusters
) it doesn't have a n_jobs
argument, which parallelizes the algorithm.
The main constraint with all of this is only being able to use the RGB and N bands as the images only contain those.
- Make sure you install or have a Python version supported by Snappy
- Install all the libraries in the requirements.txt file
- Install ESA SNAP and Snappy
- Follow all the instructions for installing Snappy in the links below
- Proceed to Running the sample
https://github.com/senbox-org/snap-engine/tree/master/snap-python/src/main/resources
- Run all the cells in cloudv1.ipynb
- Rename sample_consts.py to
consts.py
- Fill in the fields
- Uncomment line 182 in cloudv1.ipynb if you want to get your data from an S3 bucket
- Run all the cells in cloudv1.ipynb