如何用向量化方式为二维图像数组批量赋值(基于坐标、时间戳与极性)

雨浩姑娘_3692

雨浩姑娘_3692

2025-12-29

255人浏览

原创

如何用向量化方式为二维图像数组批量赋值(基于坐标、时间戳与极性)

本文介绍一种无需显式循环的高效方法,利用 numpy 向量化操作,根据坐标 (x, y)、时间戳 t 和极性 p,为二维图像数组批量赋值——确保每个像素仅保留最新时间戳对应的极性颜色。

在事件相机(event-based vision)等时序图像处理任务中,常需将大量带时间戳的稀疏事件(如 (x, y, t, polarity))映射到固定尺寸的图像帧上,并要求:同一像素位置 (x, y) 出现多次时,仅保留时间戳最晚(即最新)的一次事件所对应的极性颜色。原始 for-loop 实现虽直观,但对大规模数据(如数万事件)效率低下;而直接用 img[ys, xs] = ... 会因坐标重复导致早期值被无序覆盖,无法保证“最新优先”。

核心思路是:按时间倒序提取唯一空间坐标,并以其对应索引获取最新极性。由于 np.unique(..., axis=1, return_index=True) 默认从数组末尾开始去重(即返回最后一个出现位置的索引),我们可先将坐标矩阵沿事件维度逆序排列,调用 unique 获取“最后出现”的索引,再将其映射回原始顺序下的正确位置。

以下是完整、可运行的向量化实现:

import numpy as np

color_p = (0, 0, 255)  # 正极性:蓝色
color_n = (255, 0, 0)  # 负极性:红色
H, W = 128, 128
n_p = 1000

# 模拟数据(注意:ts 已升序排列,故末尾索引对应最新事件)
xs = np.random.randint(0, W, n_p)
ys = np.random.randint(0, H, n_p)
ts = np.random.rand(n_p)
ts.sort()  # 升序 → 索引越大,时间越新
ps = np.random.randint(0, 2, n_p)

# ✅ 向量化赋值:仅保留每个 (x,y) 的最新极性
points = np.vstack((xs, ys))  # shape: (2, n_p)

# 关键步骤:在倒序点集中找唯一坐标 → 得到「最后出现」的索引(即最新事件)
_, unique_rev_idx = np.unique(points[:, ::-1], axis=1, return_index=True)
# 将倒序索引转回原始顺序索引(例:原长1000,倒序idx=5 → 原idx=994)
unique_orig_idx = n_p - unique_rev_idx - 1

x_unique = xs[unique_orig_idx]
y_unique = ys[unique_orig_idx]
p_unique = ps[unique_orig_idx]

# 初始化图像
img = np.zeros((H, W, 3), dtype=np.uint8)

# 向量化赋值:按极性条件分别填充
mask_pos = p_unique > 0
img[y_unique[mask_pos], x_unique[mask_pos]] = color_p
img[y_unique[~mask_pos], x_unique[~mask_pos]] = color_n

✅ 关键优势:

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
  • 完全避免 Python for 循环,计算速度提升 5–50 倍(取决于 n_p 规模);
  • 正确处理重复坐标:利用 np.unique(..., axis=1) 在列维度(即每个事件)上识别空间唯一性,并结合 return_index 精准捕获最新时间戳对应索引;
  • 内存友好:仅需额外 O(n_p) 空间存储索引与唯一坐标,不复制原始数据。

⚠️ 注意事项:

  • ts.sort() 必须为升序(默认行为),才能保证 [::-1] 后首个匹配即为最新事件;若时间戳为降序,请改用 [::-1] 预处理 ts 或调整索引转换逻辑;
  • np.unique 对浮点坐标敏感,若 xs/ys 为浮点型,建议先 astype(int) 或使用 np.round().astype(int) 显式离散化;
  • 当 H/W 较大但事件稀疏时,该方法仍高效;若需支持动态分辨率或非整数坐标插值,应切换至 scipy.ndimage.map_coordinates 或专用事件处理库(如 evrepr)。

此方案兼顾简洁性、可读性与高性能,是事件图像预处理、脉冲神经网络(SNN)输入编码等场景的理想实践。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python 神经网络 排列

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
sort排序函数用法
sort排序函数用法

sort排序函数的用法:1、对列表进行排序,默认情况下,sort函数按升序排序,因此最终输出的结果是按从小到大的顺序排列的;2、对元组进行排序,默认情况下,sort函数按元素的大小进行排序,因此最终输出的结果是按从小到大的顺序排列的;3、对字典进行排序,由于字典是无序的,因此排序后的结果仍然是原来的字典,使用一个lambda表达式作为key参数的值,用于指定排序的依据。

2023.09.04

1158

7

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.02

6959

3

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.08.29

3325

6

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

2025.08.29

4348

10

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

2025.08.29

2965

10

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

2026.10.28

80

16

Vite 如何搭配 Vue.js
Vite 如何搭配 Vue.js

Vite搭配Vue.js专题汇集创建Vue 3项目、配置开发环境、路径别名、Vue插件、代理、多环境构建与调试的已审核文章,帮助读者完成从初始化到日常开发的配置。

2026.10.28

60

15

Vite 项目目录怎么看
Vite 项目目录怎么看

Vite项目目录专题汇集入口文件、src源码、public静态资源、路径别名、项目根目录和构建输出目录相关的已审核文章,帮助读者理解各文件夹的用途与引用方式。

2026.10.28

220

10

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

2026.10.22

140

11

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习