def extract_building_shadow(image, dsm, ground_spacing, radius): shadow_mask = np.zeros_like(image, dtype=np.bool) for i in range(0, image.shape[0], ground_spacing): for j in range(0, image.shape[1], ground_spacing): if not np.any(shadow_mask[i, j]): center = (i, j) ground_height = dsm[i, j] for x, y in spiral_search(center, radius, ground_spacing): if x < 0 or x >= image.shape[0] or y < 0 or y >= image.shape[1]: continue if np.any(shadow_mask[x, y:]): continue height = dsm[x, y] if height > ground_height: shadow_mask[x, y] = True elif height == ground_height: if np.linalg.norm(np.array([x, y]) - np.array(center)) < \ np.linalg.norm(np.array([i, j]) - np.array(center)): shadow_mask[x, y] = True return shadow_mask image = cv2.imread('C:\yingxiang\DJI_20230322140516_0026_V.JPG') image = cv2.resize(image, (0, 0), fx=0.1, fy=0.1) dsm_path = 'C:/sanwei/jianmo/Productions/Production_2/Production_2_DSM_part_2_2.tif' dsm_dataset = gdal.Open(dsm_path)在这段代码中如何保存掩膜结果

在这段代码中，可以使用 `cv2.imwrite` 方法将掩膜结果保存为一个新的图像文件。具体来说，可以在 `extract_building_shadow` 函数中添加以下代码，以将掩膜结果保存为一个图像文件：

# 保存掩膜结果为一个图像文件
result = np.zeros_like(image)
result[shadow_mask] = 255
cv2.imwrite("shadow_mask.png", result)

通过上述代码，将掩膜结果保存为一个图像文件 `shadow_mask.png`，其中阴影部分的像素值为 255，其余部分为 0。

需要注意的是，保存的图像文件格式可以根据需要进行修改，如 `.jpg`、`.bmp` 等。同时，也可以根据需要调整阴影部分的像素值和文件名称等参数。