python - 将 numpy 数组中的所有像素替换为单独数组中的像素,除非值为 0
问题描述
有一个我分成 3 个通道的 opencv 图像:
image #opencv image
img_red = image[:, :, 2]
img_green = image[:, :, 1]
img_blue = image[:, :, 0]
然后是三个过滤器:
red_filter
green_filter
blue_filter
它们都是 numpy 数组,但大多由零填充,因此格式如下所示:
[0, 0, 0, 132, ... 0, 15, 0, 230, 0]
...
[32, 0, 5, 0, ... 0, 2, 150, 0, 0]
我想使用这些过滤器中的每个非零值来覆盖通道中的相同索引。
像这样的东西:
img_red[index] = red_filter[index] if red_filter != 0
img_green[index] = green_filter[index] if green_filter != 0
img_blue[index] = blue_filter[index] if blue_filter != 0
final_img = cv2.merge(img_red, img_green, img_blue)
例如,如果频道看起来像这样:
[44, 225, 43, ... 24, 76, 56]
和过滤器:
[0, 0, 25 ... 2, 0, 91]
那么结果应该是:
[44, 225, 25 ... 2, 76, 91]
我试过使用 for 循环和列表推导,但是这段代码必须在视频中的每一帧上运行,所以我想知道是否有更快的方法来实现相同的结果。
在 opencv 或 numpy 操作中是否有某种图像过滤可以有效地完成这个过程?
解决方案
看来您正在寻找np.where
方法:
channel = np.array([44, 225, 43, 24, 76, 56])
filter = np.array([0, 0, 25, 2, 0, 91])
#result = np.array([44, 225, 25, 2, 76, 91])
>>> np.where(filter==0, channel, filter)
array([ 44, 225, 25, 2, 76, 91])
推荐阅读
- php - 执行代码后显示空白页
- java - 更新嵌套 JsonNode 中的值
- automated-tests - 我们可以在 cypress 的 cy.route 函数中读取并断言响应吗?
- scala - 分片编号与在 lagom 中标记事件有何关系?
- node.js - 响应输出未显示猫鼬模式 nodejs 中存在的全部信息
- c++ - 从 unsigned char 生成十六进制
- java - 如果我在批处理操作中使用 Firestore 增量,我会得到与事务相同的结果吗?
- xamarin.forms - Xamarin 通知
- java - 如何最有效地将一些信息添加到 Java 中的多个线程列表中?
- elasticsearch - 从elasticsearch查询动态分层数据