geoai库的训练数据查看与处理

1、训练数据查看

这里以官网水源数据为例：
Train water detection - geoai

运行完成后，会下载三个数据：
naip_water_train.tif(训练数据)

naip_water_test.tif（测试数据）

naip_water_masks.tif（遮罩数据），可以理解为yolo的注释数据。

通过arcgis pro软件查看这三个数据，能更直观的理解这三个数据的含义。

naip_water_train.tif(训练数据)是下图左侧的影像图，

naip_water_test.tif（测试数据）是下图右侧的影像图，

naip_water_masks.tif（遮罩数据）是下图左侧的灰色的图。

2、训练数据集

怎么准备训练数据集呢？通过上面的数据样例，可以很容易知道怎么生产训练数据集，影像底图直接下载即可。

遮罩数据可以通过shp文件生成。将处理好的shp文件导入至arcgis pro软件，然后通过要素转栅格工具转换成遮罩的tif文件，然后就有masks.tif文件。

通过geoai的算法，直接转换成所需训练集。

3、训练

训练的时候，注意使用gpu训练，方法见我上一篇博客。可以通过观察loss值，判断训练的模型效果。

4、结果预测

在arcgis pro里查看预测结果，发现基本能实现水源的预测。tif文件刚在软件里打开是黑色的，调整一下背景色为透明即可。