非AI大佬 能不能帮忙看下输出结果的栅格值为什么那么奇怪 谢谢

isdkz

输出结果的栅格值奇怪的原因是代码中没有对二值化后的图像进行数据类型转换。在创建新的输出图像时，应该将数据类型设置为gdal.GDT_Byte，即8位无符号整数。而在写入沙脊线部分时，应该将二值化后的图像数组binary转换为与输出图像相同的数据类型。修改代码如下：

1. from tifffile import imread
   
2. import numpy as np
   
3. from osgeo import gdal, osr
   
4. 
   
5. # 读取TIFF格式无人机影像数据
   
6. image_path = 'F:\duneline\dune\dune.tif'
   
7. output_path = 'overlay_image.tif'
   
8. 
   
9. image = imread(image_path)
   
10. 
    
11. # 将输入图像转换为灰度图像
    
12. gray_image = np.mean(image, axis=2)
    
13. 
    
14. # 自适应阈值二值化
    
15. binary = np.zeros_like(gray_image, dtype=np.uint8)
    
16. window_size = 131 # 窗口大小
    
17. k = 0.3 # 控制阈值的参数
    
18. for i in range(window_size//2, gray_image.shape[0]-window_size//2):
    
19.     for j in range(window_size//2, gray_image.shape[1]-window_size//2):
    
20.         window = gray_image[i-window_size//2:i+window_size//2+1, j-window_size//2:j+window_size//2+1]
    
21.         threshold = np.mean(window) - k * np.std(window)
    
22.         if gray_image[i, j] > threshold:
    
23.             binary[i, j] = 255
    
24. 
    
25. # 配置输出的空间参考信息
    
26. in_ds = gdal.Open(image_path)
    
27. projection = in_ds.GetProjection()
    
28. geotransform = in_ds.GetGeoTransform()
    
29. 
    
30. # 创建新的输出图像
    
31. driver = gdal.GetDriverByName('GTiff')
    
32. out_ds = driver.Create(output_path, image.shape[1], image.shape[0], 1, gdal.GDT_Byte)  # 修改数据类型为gdal.GDT_Byte
    
33. 
    
34. # 设置输出图像的空间参考信息
    
35. out_ds.SetProjection(projection)
    
36. out_ds.SetGeoTransform(geotransform)
    
37. 
    
38. # 将沙脊线部分写入输出图像
    
39. out_band = out_ds.GetRasterBand(1)
    
40. out_band.WriteArray(binary.astype(np.uint8))  # 将二值化后的图像数组转换为与输出图像相同的数据类型
    
41. 
    
42. # 设置颜色表
    
43. color_table = gdal.ColorTable()
    
44. color_table.SetColorEntry(0, (0, 0, 0, 0))  # 黑色，对应值为0
    
45. color_table.SetColorEntry(255, (255, 255, 255, 255))  # 白色，对应值为255
    
46. out_band.SetColorTable(color_table)
    
47. 
    
48. # 释放资源
    
49. out_band.FlushCache()
    
50. out_ds = None
    
51. in_ds = None
    
52. 
    
53. print("保存成功！")
    

复制代码

这样修改后，输出结果的栅格值应该就不会出现奇怪的情况了。