使用训练好的模型对遥感大图像进行忽略边缘（划窗）预测

大佬原文链接
大佬代码可以直接用，只需简单修改即可
依次为：要预测的遥感图像、训练好的模型、预测结果大图

预测后大图像可能全黑色，原因是像素灰度值为0或1，因此需要进行一下后处理。
下面实现代码比较老套，也可以对大图拼接代码进行修改。

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
//Result.tif为预测后的全黑大图
img = np.array(Image.open('Result.tif').convert('L'))

rows, cols = img.shape
for i in range(rows):
    for j in range(cols):
        if (img[i, j] == 1):
            img[i, j] = 255
        else:
            img[i, j] = 0

plt.figure("test")
plt.imshow(img, cmap='gray')
plt.imsave(r'C:\Users\Administrator\Desktop\1.png',img)
plt.axis('off')
plt.show()