在遥感大图 选择 小样本 训练出(深度学习)模型后,需要对整个遥感大图进行分类。通过逐块遍历,当然也可以逐像素遍历,进而将分类结果绘制在画板上。
如下为绘制代码:
def render_class2(step,size):#步长,窗口大小
tif_src = r"D:\lishihang\test_tif_peizhun_subset_proj_预测范围.tif"
dataset = gdal.Open(tif_src) # tif数据
model = load_model("D:\lishihang\model.h5") #加载模型
vec = {
'猴子': 0, '猫': 1, '狗': 2}