Skip to content

A workflow that trains a random forest/SVM machine model on given multispectral data and output the prediction of a given image in tiff format.

Notifications You must be signed in to change notification settings

LivingstoneWu/remoteSensing-images-segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 

Repository files navigation

remoteSensing-images-segmentation

1. Introduction

This project provide a workflow that trains a random forest/decision tree/SVM machine model upon given multispectral samples in xls format (which is a standard output of ArcGIS), and apply the model on a given image to output prediction result in Tiff.

2. Configuration

Required packages include GDAL, NumPy, sklearn, xlrd, pickle (only if you wish to save the model).

3. Samples

3.1 rape-wheat classification

image

A Sentinel-2 image of Yancheng, Jiangsu, China shot on 08/04/2021

The two crops are easily recognizable as the rape is flowering, showing a bright-green and yellow tone, while the wheat is dark green.
Sampling:
image

Samples took in 4 classes: water, rape, wheat, and buildings/roads

The NDVI (Normalized Difference Vegetation Index) was added into the spectral data to help the random forest model identify crops from water and buildings.

image

The classification result


Notice that, the water are identified through NDVI.

Unfortunately, since my hard disk storing these data were accidentally broken months ago, I cannot upload more examples. Later on, the same method was applied on a more virtually indistinguishable sample (wheat and maize). A CNN classifier based on UNet was also tested. Samples were generated by splitting the image into 256x256 chunks, with rotating and adding random noise.

About

A workflow that trains a random forest/SVM machine model on given multispectral data and output the prediction of a given image in tiff format.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages