如何使用 Python 在卫星图像上训练 SVM 分类器

Question

我正在使用scikit-learn 库对卫星图像执行监督分类（支持向量机分类器）。我的主要问题是如何训练我的 SVM 分类器。我在 youtube 上观看了很多视频，并阅读了一些关于如何在 scikit-learn 中训练 SVM 模型的教程。我看过的所有教程，都使用了著名的 Iris 数据集。为了在scikit-learn 中执行监督 SVM 分类，我们需要有标签。对于鸢尾花数据集，我们有Iris.target，这是我们试图预测的标签（'setosa'、'versicolor'、'virginica'）。通过阅读scikit-learn 文档，训练过程很简单。

就我而言，我必须训练在市区上空捕获的 SAR 卫星图像，并且需要对市区、道路、河流和植被进行分类（4 类）。该图像有两个波段，但我没有尝试预测的每个类别的标签数据，例如 Iris 数据。

所以，我的问题是，我是否必须手动创建向量数据（用于 4 个类）才能训练 SVM 模型？有没有比手动创建矢量数据更简单的方法来训练模型？在这种情况下我们该怎么办？

说实话，我有点困惑。我会很感激任何帮助

Answer 1

这是一个完整的示例，可以帮助您走上正轨。为简单起见，我们假设您的目标是将下面三波段图像上的像素分为三个不同的类别，即建筑物、植被和水。这些类别将分别以红色、绿色和蓝色显示。

我们首先读取图像并定义一些稍后将使用的变量。

import numpy as np
from skimage import io

img = io.imread('https://i.stack.imgur.com/TFOv7.png')

rows, cols, bands = img.shape
classes = {'building': 0, 'vegetation': 1, 'water': 2}
n_classes = len(classes)
palette = np.uint8([[255, 0, 0], [0, 255, 0], [0, 0, 255]])

无监督分类

如果您不想手动标记某些像素，那么您需要检测数据的底层结构，即您必须将图像像素拆分为n_classes 分区，例如通过k-means clustering：

from sklearn.cluster import KMeans

X = img.reshape(rows*cols, bands)
kmeans = KMeans(n_clusters=n_classes, random_state=3).fit(X)
unsupervised = kmeans.labels_.reshape(rows, cols)

io.imshow(palette[unsupervised])

监督分类

或者，您可以为某些已知类别的像素分配标签（标记像素的集合通常称为ground truth）。在这个玩具示例中，ground truth 由三个 20×20 像素的硬编码方形区域组成，如下图所示：

supervised = n_classes*np.ones(shape=(rows, cols), dtype=np.int)

supervised[200:220, 150:170] = classes['building']
supervised[40:60, 40:60] = classes['vegetation']
supervised[100:120, 200:220] = classes['water']

ground truth（训练集）的像素用于拟合支持向量机。

y = supervised.ravel()
train = np.flatnonzero(supervised < n_classes)
test = np.flatnonzero(supervised == n_classes)

from sklearn.svm import SVC

clf = SVC(gamma='auto')
clf.fit(X[train], y[train])
y[test] = clf.predict(X[test])
supervised = y.reshape(rows, cols)

io.imshow(palette[supervised])

在训练阶段之后，分类器将类标签分配给剩余的像素（测试集）。分类结果如下：

最后的评论

结果似乎表明无监督分类比有监督分类更准确。然而，监督分类通常优于无监督分类。重要的是要注意，在分析的示例中，通过调整 SVM 分类器的参数可以显着提高准确性。通过扩大和细化基本事实可以实现进一步的改进，因为训练/测试比率非常小，并且红色和绿色块实际上包含不同类别的像素。最后，人们可以合理地预期，利用更复杂的特征，例如从强度级别计算的比率或指数（例如 NDVI）会提高性能。

Answer 2

我的解决方案：-

手动处理：-

如果你的数据集很小，你可以手动创建一个矢量数据（也是可靠的，当它是你自己创建的时候）。如果不是，则很难应用 SVM 对图像进行分类。

自动处理：-

第 1 步：-

您可以使用“无监督图像聚类”技术将图像分为这 4 个类别，然后在聚类完成后将图像从 1 标记到 4。 （例如。K-Means 聚类算法）

第 2 步：-

目前，您拥有一个带标签图像的数据集。将它们拆分为训练测试数据。

第 3 步：-

现在应用 SVM 对您的测试图像进​​行分类并找出您的模型准确度。