def Read_raw(filename,w,h,m):
#初始化image
image=[]
for i in range(m*h):
image.append([])
for j in range(w):
image[i].append(255)
#读取raw
with open(filename,'rb') as f:
data=f.read()
#写入raw
for i in range(m*h):
for j in range(w):
image[i][j]=data[i*w+j]
print('读取文件{0}成功'.format(filename))
return image
def Write_raw(filename,image):
#将数组image转为bytes
bt=bytearray()
for i in range(len(image)):
for j in range(len(image[i])):
bt.append(image[i][j])
#写入raw
with open(filename,'wb') as f:
f.write(bt)
f.close()
print('写入文件{0}成功'.format(filename))
def change(image):
red=[]
green=[]
blue=[]
#遍历得像素点的最大值
L=0
for i in range(len(image)):
for j in range(len(image[i])):
if image[i][j]>L:
L=image[i][j]
print('得到像素点最大值为{0}'.format(L))
print('计算彩色图像素值中---')
for i in range(len(image)):
red.append([])
green.append([])
blue.append([])
for j in range(len(image[i])):
if image[i][j]=L/4 and image[i][j]=L/2 and image[i][j]<3*L/4:
red[i].append(4*image[i][j]-2*L)
blue[i].append(0)
green[i].append(L)
elif image[i][j]>=3*L/4 and image[i][j]<=L:
red[i].append(L)
blue[i].append(0)
green[i].append(4*L-4*image[i][j])
else:
print('error')
print('计算完成。')
#非隔行存储
colour=[]
for i in range(len(red)):
colour.append(red[i])
for i in range(len(green)):
colour.append(green[i])
for i in range(len(blue)):
colour.append(blue[i])
return colour
image=Read_raw('test2.raw',2823,2320,1)
colour=change(image)
Write_raw('test2_colour.raw',colour)
