[已解决]利用Python自动将图片转化成8bit格式

刘志佳

如题，作为一只菜鸟，不知道Python是否可以实现？想问一下大家，有没有这方面的书籍？

最佳答案

月排行榜 / 总排行榜

一个账号

2020-2-5 16:12:22

此代码转自于：https://zhuanlan.zhihu.com/p/87206303

1. import gdal
   
2. from numba import jit
   
3. import numpy as np
   
4. 
   
5. #读取tif数据集
   
6. def readTif(fileName):
   
7.     dataset = gdal.Open(fileName)
   
8.     if dataset == None:
   
9.         print(fileName+"文件无法打开")
   
10.     return dataset
    
11. 
    
12. #保存tif文件函数
    
13. def writeTiff(im_data,im_geotrans,im_proj,path):
    
14.     if 'int8' in im_data.dtype.name:
    
15.         datatype = gdal.GDT_Byte
    
16.     elif 'int16' in im_data.dtype.name:
    
17.         datatype = gdal.GDT_UInt16
    
18.     else:
    
19.         datatype = gdal.GDT_Float32
    
20.     if len(im_data.shape) == 3:
    
21.         im_bands, im_height, im_width = im_data.shape
    
22.     elif len(im_data.shape) == 2:
    
23.         im_data = np.array([im_data])
    
24.         im_bands, im_height, im_width = im_data.shape
    
25.     print(im_bands,im_width,im_height)
    
26.     print(type(im_bands),type(im_width),type(im_height))
    
27.     #创建文件
    
28.     driver = gdal.GetDriverByName("GTiff")
    
29.     dataset = driver.Create(path, int(im_width), int(im_height), int(im_bands), datatype)
    
30.     if(dataset!= None):
    
31.         dataset.SetGeoTransform(im_geotrans) #写入仿射变换参数
    
32.         dataset.SetProjection(im_proj) #写入投影
    
33.     for i in range(im_bands):
    
34.         dataset.GetRasterBand(i+1).WriteArray(im_data[i])
    
35.     del dataset
    
36.    
    
37. dataset = readTif(r"I:\gf\gs_GF2_PMS2_E97.0_N32.9_20190810_L1A0004172802.tif")
    
38. im_width = dataset.RasterXSize #栅格矩阵的列数
    
39. im_height = dataset.RasterYSize #栅格矩阵的行数
    
40. im_geotrans = dataset.GetGeoTransform()#获取仿射矩阵信息
    
41. im_proj = dataset.GetProjection()#获取投影信息
    
42. im_data = dataset.ReadAsArray(0,0,im_width,im_height)#获取数据
    
43. print(im_data.shape)
    
44. im_data_BGR = im_data[0:3,:,]
    
45. print(im_data_BGR.shape)
    
46. 
    
47. @jit
    
48. def to8bit(data):
    
49.     max_value = 0
    
50.     for i in range(data.shape[0]):
    
51.         for j in range(data.shape[1]):
    
52.             for k in range(data.shape[2]):
    
53.                 if(data[i][j][k]>max_value):
    
54.                     max_value = data[i][j][k]
    
55.     data_8bit = np.zeros(data.shape,np.uint8)
    
56.     for i in range(data.shape[0]):
    
57.         for j in range(data.shape[1]):
    
58.             for k in range(data.shape[2]):
    
59.                 data_8bit[i][j][k] = int(data[i][j][k]*255/max_value)
    
60.     return data_8bit
    
61. 
    
62. data_8bit = to8bit(im_data_BGR)
    
63. 
    
64. writeTiff(data_8bit,im_geotrans,im_proj,r"I:\gf2\8bitData\yushu_8bit.tif")

复制代码

跳转到最佳答案楼层

一个账号

此代码转自于：https://zhuanlan.zhihu.com/p/87206303

1. import gdal
   
2. from numba import jit
   
3. import numpy as np
   
4. 
   
5. #读取tif数据集
   
6. def readTif(fileName):
   
7.     dataset = gdal.Open(fileName)
   
8.     if dataset == None:
   
9.         print(fileName+"文件无法打开")
   
10.     return dataset
    
11. 
    
12. #保存tif文件函数
    
13. def writeTiff(im_data,im_geotrans,im_proj,path):
    
14.     if 'int8' in im_data.dtype.name:
    
15.         datatype = gdal.GDT_Byte
    
16.     elif 'int16' in im_data.dtype.name:
    
17.         datatype = gdal.GDT_UInt16
    
18.     else:
    
19.         datatype = gdal.GDT_Float32
    
20.     if len(im_data.shape) == 3:
    
21.         im_bands, im_height, im_width = im_data.shape
    
22.     elif len(im_data.shape) == 2:
    
23.         im_data = np.array([im_data])
    
24.         im_bands, im_height, im_width = im_data.shape
    
25.     print(im_bands,im_width,im_height)
    
26.     print(type(im_bands),type(im_width),type(im_height))
    
27.     #创建文件
    
28.     driver = gdal.GetDriverByName("GTiff")
    
29.     dataset = driver.Create(path, int(im_width), int(im_height), int(im_bands), datatype)
    
30.     if(dataset!= None):
    
31.         dataset.SetGeoTransform(im_geotrans) #写入仿射变换参数
    
32.         dataset.SetProjection(im_proj) #写入投影
    
33.     for i in range(im_bands):
    
34.         dataset.GetRasterBand(i+1).WriteArray(im_data[i])
    
35.     del dataset
    
36.    
    
37. dataset = readTif(r"I:\gf\gs_GF2_PMS2_E97.0_N32.9_20190810_L1A0004172802.tif")
    
38. im_width = dataset.RasterXSize #栅格矩阵的列数
    
39. im_height = dataset.RasterYSize #栅格矩阵的行数
    
40. im_geotrans = dataset.GetGeoTransform()#获取仿射矩阵信息
    
41. im_proj = dataset.GetProjection()#获取投影信息
    
42. im_data = dataset.ReadAsArray(0,0,im_width,im_height)#获取数据
    
43. print(im_data.shape)
    
44. im_data_BGR = im_data[0:3,:,]
    
45. print(im_data_BGR.shape)
    
46. 
    
47. @jit
    
48. def to8bit(data):
    
49.     max_value = 0
    
50.     for i in range(data.shape[0]):
    
51.         for j in range(data.shape[1]):
    
52.             for k in range(data.shape[2]):
    
53.                 if(data[i][j][k]>max_value):
    
54.                     max_value = data[i][j][k]
    
55.     data_8bit = np.zeros(data.shape,np.uint8)
    
56.     for i in range(data.shape[0]):
    
57.         for j in range(data.shape[1]):
    
58.             for k in range(data.shape[2]):
    
59.                 data_8bit[i][j][k] = int(data[i][j][k]*255/max_value)
    
60.     return data_8bit
    
61. 
    
62. data_8bit = to8bit(im_data_BGR)
    
63. 
    
64. writeTiff(data_8bit,im_geotrans,im_proj,r"I:\gf2\8bitData\yushu_8bit.tif")

复制代码