python数组溢出_Python，OpenCV：增加图像亮度而不会溢出UINT8数组

最新推荐文章于 2022-04-22 20:20:53 发布

weixin_39521009

最新推荐文章于 2022-04-22 20:20:53 发布

阅读量116

点赞数

文章标签： python数组溢出

I am trying to increase brightness of a grayscale image. cv2.imread() returns a numpy array. I am adding integer value to every element of the array. Theoretically, this would increase each of them. After that I would be able to put upper threshold of 255 and get the image with the higher brightness.

Here is the code:

grey = cv2.imread(path+file,0)

print type(grey)

print grey[0]

new = grey + value

print new[0]

res = np.hstack((grey, new))

cv2.imshow('image', res)

cv2.waitKey(0)

cv2.destroyAllWindows()

However, internal OpenCV routine apparently does something like that:

new_array = old_array % 255

Every pixel intensity value higher than 255 becomes a remainder of dividing by 255.

As a result, I am getting dark instead of completely white.

Here is the output:

[115 114 121 ..., 170 169 167]

[215 214 221 ..., 14 13 11]

And here is the image:

How can I switch off this remainder mechanism? Is there any better way to increase brightness in OpenCV?

解决方案

One idea would be to check before adding value whether the addition would result in an overflow by checking the difference between 255 and the current pixel value and checking if it's within value. If it does, we won't add value, we would directly set those at 255, otherwise we would do the addition. Now, this decision making could be eased up with a mask creation and would be -

mask = (255 - grey) < value

Then, feed this mask/boolean array to np.where to let it choose between 255 and grey+value based on the mask.

Thus, finally we would have the implementation as -

grey_new = np.where((255 - grey) < value,255,grey+value)

Sample run

Let's use a small representative example to demonstrate the steps.

In [340]: grey

Out[340]:

array([[125, 212, 104, 180, 244],

[105, 26, 132, 145, 157],

[126, 230, 225, 204, 91],

[226, 181, 43, 122, 125]], dtype=uint8)