np.where在处理图像像素时需要注意的小问题

np.where在处理图像像素时需要注意的小问题

  • 1.问题描述
  • 2.问题分析
  • 3.解决办法

1.问题描述

因为只想对像素值满足一定条件的像素做修改,所以用到了Numpy中的where方法,具体内容见上篇文章(https://editor.csdn.net/md/?articleId=127509748)。

现在想给满足条件的像素值加上delta,如果加上delta的值大于255,就将像素点值设置为255,如果小于0,就设置为0.在实验过程发现,当加上delta后的值大于255时,并没有将该点值设置为255,以一个3*3的矩阵为例:
因为图像每个通道元素的位深一般为uint8,这里为了模拟图像的某个通道的矩阵,创建矩阵时申明dtype=np.uint8

delta=250
L1=np.array([[0,20,40],[3,14,4],[1,30,15]],dtype=np.uint8)
a1=np.where(L1+delta>255 ,255,0)

结果如下:

[[0 0 0]
 [0 0 0]
 [0 0 0]]

发现并没有把L1+delta>255的值设置为255。

2.问题分析

这主要跟L1元素的位深度有关系,uint8的范围是0~255,如果和的结果大于255时,结果会除255取余,通俗理解就是超过255后从0再开始加。这样L1+delta的值永远不会大于255的。

3.解决办法

将L1的位深先改变位uint16,运算结束后再改为原来的uint8(如果需要的话).

import numpy as np
import cv2
delta=250
L1=np.array([[0,20,40],[3,14,4],[1,30,15]],dtype=np.uint8)
L1=np.uint16(L1)
a1=np.where(L1+delta>255 ,255,0)
print(a1)

输出结果:

[[  0 255 255]
 [  0 255   0]
 [  0 255 255]]

可以看到原像素值加delta大于255时已经全部设置为255,问题解决!

你可能感兴趣的:(小问题记录,numpy,python)