反转numpy数组图像,可能是uint8,uint16,

时间:2013-06-26 21:15:57

标签: python image-processing numpy

有更好的方法吗?具体来说,这些最大值是否可通过numpy API获得?我无法在API中找到它们,尽管很容易找到here in the docs

MAX_VALUES = {np.uint8: 255, np.uint16: 65535, np.uint32: 4294967295, \
              np.uint64: 18446744073709551615}

try:
    image = MAX_VALUES[image.dtype] - image
except KeyError:
    raise ValueError, "Image must be array of unsigned integers."

像PIL和cv2这样的软件包为反转图像提供了方便的工具,但是在代码的这一点上我有一个numpy数组 - 接下来是更复杂的分析 - 我想坚持使用numpy。

2 个答案:

答案 0 :(得分:3)

尝试

image ^= MAX_VALUES[image.dtype]

答案 1 :(得分:1)

顺便说一句,您不需要自己定义MAX_VALUES。 NumPy has them built-in

import numpy as np
h, w = 100, 100
image = np.arange(h*w).reshape((h,w)).astype(np.uint8)
max_val = np.iinfo(image.dtype).max
print(max_val)
# 255
image ^= max_val

print(image)
# [[255 254 253 ..., 158 157 156]
#  [155 154 153 ...,  58  57  56]
#  [ 55  54  53 ..., 214 213 212]
#  ..., 
#  [ 27  26  25 ..., 186 185 184]
#  [183 182 181 ...,  86  85  84]
#  [ 83  82  81 ..., 242 241 240]]