之前写过一篇使用R语言对卫星影像进行kmeans聚类的文章,本文来个python版本的。
Python版的没有R语言版本的简单(代码多了一些),但是通过Python版的学习可以清楚了解到对卫星影像进行分类的整个流程。因为Python在机器学习/深度学习方面比较通用,学会了本文之后,就学会了任何基于机器学习对卫星影像进行分类的流程,只需要换下中间的分类算法即可。
公众号《GIS与Climate》,分享R、Python、GIS和Climate的相关技术,求关注。
导入需要的包
因为本文用的是kmeans算法,也没什么高大上的,需要的包也就比较少:
import rasterio as rio
from rasterio.plot import show
from sklearn import cluster
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.colors as mc
读取数据
我们用rasterio包来读取栅格数据(这里以tif格式作为示例):