1. 端侧视觉 AI 的工程真相为什么要把神经网络塞进浏览器标签页第一次听到“把神经网络塞进浏览器标签页”这个说法很多人脑子里浮现的画面可能是打开一个网页摄像头一开框框就自动画出来了全程不联网也不装任何客户端。听起来像魔法但拆开看它其实是一堆相当朴素的工程取舍堆出来的结果。我最早接触这类需求是帮一个做工业质检的朋友做原型验证——他们的产线相机拍到的画面涉及工艺细节客户明确要求“数据不出本地”但同时又希望现场工程师用最轻的方式就能跑起来不想装驱动、不想配环境、不想维护一堆 Python 依赖。浏览器恰好是那个“每个工位都有的运行时”。端侧视觉 AI 的核心诉求说白了就三件事数据不出端、延迟足够低、部署足够轻。把神经网络放进浏览器标签页本质上是用 Web 这个跨平台运行时去承载原本属于本地客户端的推理任务。它解决的不是“模型精度”问题而是“最后一公里交付”问题。适合谁来参考如果你是会写前端、但对模型部署一知半解的同学这篇能帮你把链路打通如果你是做算法的、想把 demo 快速给客户看这篇能帮你避开浏览器这个“坑王”里最常见的几个雷。我下面讲的东西全部围绕一个真实可复现的链路展开摄像头采集 → Web Worker 里跑推理 → WebGL 做张量加速 → 主线程只负责画框。中间会穿插大量我踩过的坑比如为什么你的模型在 Chrome 上跑得好好的到了某些浏览器就卡成 PPT为什么 Web Worker 里拿不到 DOM却反而成了性能救星为什么 WebGL 的坐标系和 HTML 的坐标系差了一个 Y 轴害我调了一下午的框位置。2. 整体架构设计与方案选型为什么是 Worker WebGL 这套组合2.1 从“主线程跑模型”到“Worker 里跑模型”的必然性最朴素的实现方式是在主线程里直接加载模型、直接推理。我一开始就是这么干的结果非常直观页面直接卡死。因为 JavaScript 是单线程的主线程既要处理 DOM 渲染、又要处理用户交互、还要跑几十毫秒甚至上百毫秒一次的推理浏览器根本忙不过来。你看到的画面就是摄像头预览一顿一顿鼠标点哪都没反应标签页甚至会被浏览器判定为“无响应”。所以第一刀必须切在线程模型上。Web Worker 的价值在这里体现得淋漓尽致它跑在独立的线程里不阻塞 UI。主线程只做两件事——把摄像头帧丢给 Worker以及接收 Worker 返回的检测结果去画框。Worker 里则专心做预处理、推理、后处理。这个分工一旦定下来整个页面的流畅度会有质的提升。但 Worker 有个硬限制它不能访问 DOM也不能直接操作 Canvas 的 2D 上下文。这意味着你不能在 Worker 里直接画框。很多人第一次遇到这个限制会觉得很别扭但换个角度想这恰恰是它性能好的原因——它不被渲染任务干扰。我的做法是Worker 只返回结构化的检测结果比如[{x, y, w, h, score, class}]主线程拿到后在一个覆盖在视频上的 Canvas 里画框。职责清晰互不干扰。2.2 WebGL 为什么是浏览器里跑神经网络的“加速器”光有 Worker 还不够。纯 JavaScript 做卷积运算速度慢到你想哭。一个中等规模的卷积神经网络用 JS 逐元素算单帧可能要几百毫秒甚至上秒。这时候就轮到WebGL上场了。WebGL 的本质是让 JavaScript 能调用 GPU 做并行计算。神经网络的卷积、矩阵乘法天然就是高度并行的任务非常适合丢给 GPU。你不需要自己写 shader 去实现每一个算子现在主流的浏览器端推理框架比如 ONNX Runtime Web、TensorFlow.js 的 WebGL 后端已经把常用算子用 GLSL 实现好了。你只需要在初始化时指定后端为webgl它就会自动把张量运算搬到 GPU 上。这里有个关键取舍WebGL 有精度限制。它默认用的是 32 位浮点纹理但很多移动端 GPU 对浮点纹理的支持并不完整可能退化成 16 位甚至更低。这会导致模型输出和桌面端有细微差异。如果你的模型对数值精度极其敏感比如某些回归任务建议在 WebGL 后端跑完后对关键结果做一次 CPU 侧的校验或修正。我做过一个姿态估计的项目WebGL 后端输出的关键点坐标在小数点后两位会有抖动后来加了一层简单的滑动平均滤波才稳定下来。2.3 为什么不用 WebAssembly 或 WebGPU有人会问现在不是有 WebAssembly 和 WebGPU 吗为什么还用 WebGL我的回答是看场景。WebAssembly 的优势是能跑 C 编译过来的代码性能接近原生但它本身不直接调用 GPU。如果你用 WASM 跑纯 CPU 推理速度比 WebGL 慢一个数量级。除非你的模型非常小或者你用的是 WASM SIMD 的优化版本否则在视觉任务上WebGL 仍然是更实际的选择。WebGPU 是未来它的计算着色器比 WebGL 更灵活、性能更好但兼容性还在爬坡。截至我写这篇的时候部分浏览器版本对 WebGPU 的支持仍然需要手动开启 flag移动端更是参差不齐。如果你的项目要求“打开就能用”WebGL 是当下最稳的赌注。我的建议是架构上把推理后端抽象成一个接口先上 WebGL等 WebGPU 普及了再平滑切换。3. 核心细节解析与实操要点从摄像头到检测框的完整链路3.1 摄像头采集别小看 getUserMedia 的坑采集这一步API 本身很简单navigator.mediaDevices.getUserMedia({ video: true })。但实际用起来坑不少。第一个坑是分辨率与帧率的权衡。你请求 1920x1080浏览器不一定给你。它会根据设备能力和当前负载协商一个最接近的值。我建议在getUserMedia的 constraints 里明确写上你期望的分辨率然后在video元素的loadedmetadata事件里读取实际拿到的videoWidth和videoHeight用实际值去做后续的坐标映射。否则你的检测框会莫名其妙偏移。第二个坑是自动曝光和自动对焦。工业场景下环境光变化会导致画面忽明忽暗模型输入分布漂移检测结果就会跳。如果设备支持可以通过applyConstraints关掉自动曝光手动锁定曝光值。这个在消费级摄像头上不一定生效但值得一试。第三个坑是权限与安全上下文。getUserMedia只在 HTTPS 或 localhost 下可用。如果你在局域网里用 HTTP 访问摄像头根本起不来。这不是 bug是浏览器的安全策略。解决办法要么上 HTTPS要么在本地起服务。3.2 预处理把视频帧变成模型能吃的张量模型不会直接吃video元素。你需要把当前帧画到一个离屏 Canvas 上然后读出像素数据再转换成模型需要的张量格式。这一步的细节决定了你的推理速度。我的做法是在 Worker 里创建一个OffscreenCanvas把VideoFrame或ImageBitmap画上去然后用getImageData拿到Uint8ClampedArray。接着做归一化比如除以 255、减均值、除标准差最后按 NCHW 或 NHWC 排布成Float32Array。这个过程听起来简单但每一步都有优化空间。比如归一化如果你在 JS 里逐像素做(pixel / 255 - mean) / std一帧 640x640 的图就是 122 万次运算虽然不算多但累积起来也可观。更快的做法是把归一化参数直接写进模型的预处理层或者用 WebGL 的 shader 做。我实测下来把预处理也搬到 GPU 上整体延迟能降 15% 到 20%。还有一个容易忽略的点颜色空间。摄像头输出通常是 YUV浏览器帮你转成了 RGB但这个转换在不同浏览器上可能有细微差异。如果你的模型对颜色敏感比如做颜色分类建议在预处理时统一做一次颜色空间转换确保输入一致。3.3 推理模型加载、会话创建与后端选择模型加载这一步我强烈建议用ONNX Runtime Web或TensorFlow.js这类成熟框架不要自己手写算子。原因很简单它们已经帮你处理了 WebGL 后端的各种兼容性问题包括纹理格式、精度回退、内存管理等。你自己写光是一个卷积的 shader 就能调一周。以 ONNX Runtime Web 为例初始化流程大致是先ort.env.wasm.wasmPaths指定 WASM 文件路径如果你用了 WASM 后端然后ort.InferenceSession.create(modelUrl, { executionProviders: [webgl] })。这里有个细节executionProviders 的顺序很重要。如果你写[webgl, wasm]它会优先尝试 WebGL失败后回退到 WASM。这个回退机制在兼容性上非常有用但你要知道回退发生后性能会下降最好在日志里打出来。模型本身的选择也有讲究。浏览器端不适合跑太大的模型。我的经验是参数量控制在 5M 以内输入分辨率控制在 640x640 以内这样在主流笔记本上能做到 20 到 30 FPS。如果你非要跑大模型那就得接受帧率下降或者做模型量化。量化到 INT8 能显著减小模型体积、提升速度但精度损失需要你自己评估。我做过一个对比同一个检测模型FP32 在 WebGL 上跑 18 FPSINT8 能跑到 35 FPS但小目标的召回率掉了大约 3 个百分点。这个取舍要看你的业务能不能接受。3.4 后处理NMS 与坐标映射模型输出通常是一堆候选框和分数你需要做非极大值抑制NMS来去重。NMS 本身是纯计算放在 Worker 里用 JS 做就行不需要上 GPU。但要注意NMS 的阈值选择直接影响体验。阈值太高框会重复阈值太低相邻物体容易被误删。我一般从 0.45 开始调根据实际场景微调。坐标映射是另一个高频出错点。模型输入的坐标是相对于输入张量的比如 640x640而你要画框的 Canvas 是相对于视频显示区域的。这中间涉及缩放、裁剪、偏移三种变换。如果你在预处理时做了 letterbox保持宽高比填充那后处理时就要把填充的边距减掉再按比例映射回原始视频尺寸。我见过太多人在这里搞错导致框整体偏移或者大小不对。建议写一个独立的坐标变换函数输入输出都用明确的坐标系定义避免混淆。4. 实操过程与核心环节实现一个可复现的端侧检测 Demo4.1 环境准备与项目结构我假设你用的是现代前端工具链比如 Vite。项目结构大概是这样project/ public/ model.onnx ort-wasm-simd.wasm src/ main.js worker.js index.htmlmain.js负责 UI 和摄像头采集worker.js负责推理。两者通过postMessage通信。模型文件放在public下确保能被直接访问。4.2 主线程采集与渲染主线程的核心逻辑是启动摄像头创建一个requestAnimationFrame循环每隔一定帧数把当前视频帧转成ImageBitmap发给 Worker。为什么不每帧都发因为推理速度可能跟不上摄像头帧率发太快只会堆积任务。我的做法是用“上一帧处理完再发下一帧”的背压机制保证不会积压。const video document.getElementById(video); const canvas document.getElementById(overlay); const ctx canvas.getContext(2d); const worker new Worker(./worker.js, { type: module }); let busy false; async function loop() { if (!busy video.readyState 2) { busy true; const bitmap await createImageBitmap(video); worker.postMessage({ type: frame, bitmap }, [bitmap]); } requestAnimationFrame(loop); } worker.onmessage (e) { const { boxes } e.data; ctx.clearRect(0, 0, canvas.width, canvas.height); for (const box of boxes) { ctx.strokeStyle #00ff00; ctx.lineWidth 2; ctx.strokeRect(box.x, box.y, box.w, box.h); } busy false; }; loop();注意createImageBitmap是异步的而且它创建的 bitmap 是可转移对象通过postMessage的第二个参数转移给 Worker 后主线程就不再持有它避免了拷贝开销。这个小技巧能省不少内存带宽。4.3 Worker推理与后处理Worker 里初始化 ONNX Runtime 会话然后监听消息。收到帧后画到OffscreenCanvas取像素做预处理跑推理做 NMS最后把框发回去。import * as ort from onnxruntime-web; let session null; const INPUT_SIZE 640; async function init() { ort.env.wasm.wasmPaths /; session await ort.InferenceSession.create(/model.onnx, { executionProviders: [webgl, wasm], graphOptimizationLevel: all }); } self.onmessage async (e) { if (e.data.type frame) { const bitmap e.data.bitmap; const canvas new OffscreenCanvas(INPUT_SIZE, INPUT_SIZE); const ctx canvas.getContext(2d); ctx.drawImage(bitmap, 0, 0, INPUT_SIZE, INPUT_SIZE); const imageData ctx.getImageData(0, 0, INPUT_SIZE, INPUT_SIZE); const input preprocess(imageData); const tensor new ort.Tensor(float32, input, [1, 3, INPUT_SIZE, INPUT_SIZE]); const outputs await session.run({ images: tensor }); const boxes postprocess(outputs, bitmap.width, bitmap.height); self.postMessage({ boxes }); bitmap.close(); } };preprocess里做归一化和 HWC 到 CHW 的转换。postprocess里做 NMS 和坐标映射。这两个函数是性能热点建议用Float32Array和for循环避免高阶数组方法。4.4 参数计算输入尺寸与坐标映射的数学假设原始视频是 1280x720模型输入是 640x640。如果你直接拉伸宽高比就变了检测框会变形。所以要用 letterbox先算缩放比例scale min(640/1280, 640/720) 0.5然后缩放后尺寸是 640x360上下各填充(640-360)/2 140像素。后处理时模型输出的框坐标是在 640x640 空间里的。你要先减去填充量y y - 140再除以缩放比例y y / 0.5。这样得到的才是原始视频坐标系下的框。x 方向同理但因为宽度刚好是 640没有填充所以x x / 0.5。这个计算过程我建议写成独立函数并且用单元测试验证。我当初就是靠一个简单的测试用例已知输入输出才发现自己把填充量加错了方向。5. 常见问题与排查技巧实录那些文档里不会写的坑5.1 性能突然下降WebGL 后端悄悄回退了这是最隐蔽的问题之一。你明明指定了webgl但性能就是上不去。原因可能是GPU 不支持某个算子框架自动回退到了 WASM。排查方法是打开浏览器的开发者工具看控制台有没有回退警告或者在代码里监听session的后端信息。如果发现回退了要么换一个算子兼容性更好的模型要么接受 WASM 的性能。另一个性能杀手是纹理上传开销。每一帧都要把图像数据传到 GPU如果图像很大这个传输时间可能比推理本身还长。优化方法是尽量在 GPU 侧完成预处理减少 CPU 和 GPU 之间的数据往返。5.2 内存泄漏Worker 里的张量没释放ONNX Runtime Web 的Tensor对象如果频繁创建而不释放内存会持续增长。虽然它有垃圾回收但在高频推理场景下GC 跟不上分配速度。我的做法是复用输入张量的缓冲区每次只更新数据不重新创建对象。输出张量也要及时置空让 GC 能回收。还有一个容易忽略的点ImageBitmap用完必须调close()。我见过一个项目跑了几分钟后标签页直接崩溃就是因为 bitmap 没释放内存爆了。5.3 跨浏览器差异Chrome 能跑别的浏览器不行不同浏览器对 WebGL 扩展的支持不一样。比如OES_texture_float这个扩展有些浏览器支持有些不支持。如果不支持浮点纹理就用不了推理精度和速度都会受影响。我的建议是在初始化时检测关键扩展如果不支持就降级到 WASM 后端并给用户一个提示。另外Safari 对 WebGL 的实现有一些自己的“脾气”比如对纹理尺寸有限制超过一定大小会静默失败。如果你发现 Safari 上模型加载失败先检查输入尺寸是不是太大了。5.4 常见问题速查表问题现象可能原因排查方法解决思路页面卡死推理在主线程跑看 Performance 面板移到 Web Worker帧率低WebGL 回退到 WASM看控制台警告换模型或接受降级检测框偏移坐标映射错误打印中间坐标检查 letterbox 参数内存持续增长张量或 bitmap 未释放看内存快照复用缓冲区及时 close摄像头打不开非 HTTPS 环境看控制台报错上 HTTPS 或 localhost模型加载失败路径或格式错误看网络请求检查 MIME 类型和路径5.5 独家避坑技巧第一个技巧在 Worker 里做推理时把session.run包在 try-catch 里。因为 GPU 上下文可能因为各种原因丢失比如系统休眠恢复后一旦丢失后续推理全部失败。捕获到错误后重新创建 session 就能恢复。第二个技巧用performance.now()打点把预处理、推理、后处理的时间分别记录下来。这样你才知道瓶颈在哪。我一开始以为推理最慢结果发现预处理里的getImageData才是大头后来改用OffscreenCanvas的transferToImageBitmap才优化下来。第三个技巧模型输入尺寸不要盲目追求大。640 和 416 的精度差异在很多场景下肉眼几乎看不出但速度差了一倍。先用小尺寸验证链路再根据精度需求往上调。6. 端侧视觉 AI 的边界与我的实际体会把神经网络塞进浏览器标签页听起来很酷但它有明确的边界。它不适合跑超大模型不适合对延迟要求极致到毫秒级的场景也不适合需要复杂多模型串联的任务。它的甜点区是中等规模模型、实时性要求 20 FPS 以上、部署环境不可控、数据不能出端。工业质检、教育演示、隐私敏感的消费应用都是它的用武之地。我在实际项目里最大的体会是浏览器端的性能优化八成功夫在数据搬运上。模型本身的计算框架已经帮你优化得很好了。真正拖慢速度的是图像从摄像头到 CPU、从 CPU 到 GPU、从 Worker 到主线程的这些搬运过程。谁能把这些搬运做到最少、最快谁就能把帧率提上去。最后分享一个小技巧如果你的场景允许把视频帧直接以VideoFrame的形式传给 Worker而不是转成ImageBitmap。VideoFrame是更底层的对象某些浏览器上它的传输开销更小。不过兼容性需要你自己测一下不是所有浏览器都支持。这个方向后续还可以往 WebGPU 迁移等兼容性成熟后计算着色器能带来更灵活的算子实现和更好的性能。但在那之前WebGL Worker 这套组合仍然是浏览器端视觉 AI 最稳的工程方案。
1. 端侧视觉 AI 的工程真相为什么要把神经网络塞进浏览器标签页第一次听到“把神经网络塞进浏览器标签页”这个说法很多人脑子里浮现的画面可能是打开一个网页摄像头一开框框就自动画出来了全程不联网也不装任何客户端。听起来像魔法但拆开看它其实是一堆相当朴素的工程取舍堆出来的结果。我最早接触这类需求是帮一个做工业质检的朋友做原型验证——他们的产线相机拍到的画面涉及工艺细节客户明确要求“数据不出本地”但同时又希望现场工程师用最轻的方式就能跑起来不想装驱动、不想配环境、不想维护一堆 Python 依赖。浏览器恰好是那个“每个工位都有的运行时”。端侧视觉 AI 的核心诉求说白了就三件事数据不出端、延迟足够低、部署足够轻。把神经网络放进浏览器标签页本质上是用 Web 这个跨平台运行时去承载原本属于本地客户端的推理任务。它解决的不是“模型精度”问题而是“最后一公里交付”问题。适合谁来参考如果你是会写前端、但对模型部署一知半解的同学这篇能帮你把链路打通如果你是做算法的、想把 demo 快速给客户看这篇能帮你避开浏览器这个“坑王”里最常见的几个雷。我下面讲的东西全部围绕一个真实可复现的链路展开摄像头采集 → Web Worker 里跑推理 → WebGL 做张量加速 → 主线程只负责画框。中间会穿插大量我踩过的坑比如为什么你的模型在 Chrome 上跑得好好的到了某些浏览器就卡成 PPT为什么 Web Worker 里拿不到 DOM却反而成了性能救星为什么 WebGL 的坐标系和 HTML 的坐标系差了一个 Y 轴害我调了一下午的框位置。2. 整体架构设计与方案选型为什么是 Worker WebGL 这套组合2.1 从“主线程跑模型”到“Worker 里跑模型”的必然性最朴素的实现方式是在主线程里直接加载模型、直接推理。我一开始就是这么干的结果非常直观页面直接卡死。因为 JavaScript 是单线程的主线程既要处理 DOM 渲染、又要处理用户交互、还要跑几十毫秒甚至上百毫秒一次的推理浏览器根本忙不过来。你看到的画面就是摄像头预览一顿一顿鼠标点哪都没反应标签页甚至会被浏览器判定为“无响应”。所以第一刀必须切在线程模型上。Web Worker 的价值在这里体现得淋漓尽致它跑在独立的线程里不阻塞 UI。主线程只做两件事——把摄像头帧丢给 Worker以及接收 Worker 返回的检测结果去画框。Worker 里则专心做预处理、推理、后处理。这个分工一旦定下来整个页面的流畅度会有质的提升。但 Worker 有个硬限制它不能访问 DOM也不能直接操作 Canvas 的 2D 上下文。这意味着你不能在 Worker 里直接画框。很多人第一次遇到这个限制会觉得很别扭但换个角度想这恰恰是它性能好的原因——它不被渲染任务干扰。我的做法是Worker 只返回结构化的检测结果比如[{x, y, w, h, score, class}]主线程拿到后在一个覆盖在视频上的 Canvas 里画框。职责清晰互不干扰。2.2 WebGL 为什么是浏览器里跑神经网络的“加速器”光有 Worker 还不够。纯 JavaScript 做卷积运算速度慢到你想哭。一个中等规模的卷积神经网络用 JS 逐元素算单帧可能要几百毫秒甚至上秒。这时候就轮到WebGL上场了。WebGL 的本质是让 JavaScript 能调用 GPU 做并行计算。神经网络的卷积、矩阵乘法天然就是高度并行的任务非常适合丢给 GPU。你不需要自己写 shader 去实现每一个算子现在主流的浏览器端推理框架比如 ONNX Runtime Web、TensorFlow.js 的 WebGL 后端已经把常用算子用 GLSL 实现好了。你只需要在初始化时指定后端为webgl它就会自动把张量运算搬到 GPU 上。这里有个关键取舍WebGL 有精度限制。它默认用的是 32 位浮点纹理但很多移动端 GPU 对浮点纹理的支持并不完整可能退化成 16 位甚至更低。这会导致模型输出和桌面端有细微差异。如果你的模型对数值精度极其敏感比如某些回归任务建议在 WebGL 后端跑完后对关键结果做一次 CPU 侧的校验或修正。我做过一个姿态估计的项目WebGL 后端输出的关键点坐标在小数点后两位会有抖动后来加了一层简单的滑动平均滤波才稳定下来。2.3 为什么不用 WebAssembly 或 WebGPU有人会问现在不是有 WebAssembly 和 WebGPU 吗为什么还用 WebGL我的回答是看场景。WebAssembly 的优势是能跑 C 编译过来的代码性能接近原生但它本身不直接调用 GPU。如果你用 WASM 跑纯 CPU 推理速度比 WebGL 慢一个数量级。除非你的模型非常小或者你用的是 WASM SIMD 的优化版本否则在视觉任务上WebGL 仍然是更实际的选择。WebGPU 是未来它的计算着色器比 WebGL 更灵活、性能更好但兼容性还在爬坡。截至我写这篇的时候部分浏览器版本对 WebGPU 的支持仍然需要手动开启 flag移动端更是参差不齐。如果你的项目要求“打开就能用”WebGL 是当下最稳的赌注。我的建议是架构上把推理后端抽象成一个接口先上 WebGL等 WebGPU 普及了再平滑切换。3. 核心细节解析与实操要点从摄像头到检测框的完整链路3.1 摄像头采集别小看 getUserMedia 的坑采集这一步API 本身很简单navigator.mediaDevices.getUserMedia({ video: true })。但实际用起来坑不少。第一个坑是分辨率与帧率的权衡。你请求 1920x1080浏览器不一定给你。它会根据设备能力和当前负载协商一个最接近的值。我建议在getUserMedia的 constraints 里明确写上你期望的分辨率然后在video元素的loadedmetadata事件里读取实际拿到的videoWidth和videoHeight用实际值去做后续的坐标映射。否则你的检测框会莫名其妙偏移。第二个坑是自动曝光和自动对焦。工业场景下环境光变化会导致画面忽明忽暗模型输入分布漂移检测结果就会跳。如果设备支持可以通过applyConstraints关掉自动曝光手动锁定曝光值。这个在消费级摄像头上不一定生效但值得一试。第三个坑是权限与安全上下文。getUserMedia只在 HTTPS 或 localhost 下可用。如果你在局域网里用 HTTP 访问摄像头根本起不来。这不是 bug是浏览器的安全策略。解决办法要么上 HTTPS要么在本地起服务。3.2 预处理把视频帧变成模型能吃的张量模型不会直接吃video元素。你需要把当前帧画到一个离屏 Canvas 上然后读出像素数据再转换成模型需要的张量格式。这一步的细节决定了你的推理速度。我的做法是在 Worker 里创建一个OffscreenCanvas把VideoFrame或ImageBitmap画上去然后用getImageData拿到Uint8ClampedArray。接着做归一化比如除以 255、减均值、除标准差最后按 NCHW 或 NHWC 排布成Float32Array。这个过程听起来简单但每一步都有优化空间。比如归一化如果你在 JS 里逐像素做(pixel / 255 - mean) / std一帧 640x640 的图就是 122 万次运算虽然不算多但累积起来也可观。更快的做法是把归一化参数直接写进模型的预处理层或者用 WebGL 的 shader 做。我实测下来把预处理也搬到 GPU 上整体延迟能降 15% 到 20%。还有一个容易忽略的点颜色空间。摄像头输出通常是 YUV浏览器帮你转成了 RGB但这个转换在不同浏览器上可能有细微差异。如果你的模型对颜色敏感比如做颜色分类建议在预处理时统一做一次颜色空间转换确保输入一致。3.3 推理模型加载、会话创建与后端选择模型加载这一步我强烈建议用ONNX Runtime Web或TensorFlow.js这类成熟框架不要自己手写算子。原因很简单它们已经帮你处理了 WebGL 后端的各种兼容性问题包括纹理格式、精度回退、内存管理等。你自己写光是一个卷积的 shader 就能调一周。以 ONNX Runtime Web 为例初始化流程大致是先ort.env.wasm.wasmPaths指定 WASM 文件路径如果你用了 WASM 后端然后ort.InferenceSession.create(modelUrl, { executionProviders: [webgl] })。这里有个细节executionProviders 的顺序很重要。如果你写[webgl, wasm]它会优先尝试 WebGL失败后回退到 WASM。这个回退机制在兼容性上非常有用但你要知道回退发生后性能会下降最好在日志里打出来。模型本身的选择也有讲究。浏览器端不适合跑太大的模型。我的经验是参数量控制在 5M 以内输入分辨率控制在 640x640 以内这样在主流笔记本上能做到 20 到 30 FPS。如果你非要跑大模型那就得接受帧率下降或者做模型量化。量化到 INT8 能显著减小模型体积、提升速度但精度损失需要你自己评估。我做过一个对比同一个检测模型FP32 在 WebGL 上跑 18 FPSINT8 能跑到 35 FPS但小目标的召回率掉了大约 3 个百分点。这个取舍要看你的业务能不能接受。3.4 后处理NMS 与坐标映射模型输出通常是一堆候选框和分数你需要做非极大值抑制NMS来去重。NMS 本身是纯计算放在 Worker 里用 JS 做就行不需要上 GPU。但要注意NMS 的阈值选择直接影响体验。阈值太高框会重复阈值太低相邻物体容易被误删。我一般从 0.45 开始调根据实际场景微调。坐标映射是另一个高频出错点。模型输入的坐标是相对于输入张量的比如 640x640而你要画框的 Canvas 是相对于视频显示区域的。这中间涉及缩放、裁剪、偏移三种变换。如果你在预处理时做了 letterbox保持宽高比填充那后处理时就要把填充的边距减掉再按比例映射回原始视频尺寸。我见过太多人在这里搞错导致框整体偏移或者大小不对。建议写一个独立的坐标变换函数输入输出都用明确的坐标系定义避免混淆。4. 实操过程与核心环节实现一个可复现的端侧检测 Demo4.1 环境准备与项目结构我假设你用的是现代前端工具链比如 Vite。项目结构大概是这样project/ public/ model.onnx ort-wasm-simd.wasm src/ main.js worker.js index.htmlmain.js负责 UI 和摄像头采集worker.js负责推理。两者通过postMessage通信。模型文件放在public下确保能被直接访问。4.2 主线程采集与渲染主线程的核心逻辑是启动摄像头创建一个requestAnimationFrame循环每隔一定帧数把当前视频帧转成ImageBitmap发给 Worker。为什么不每帧都发因为推理速度可能跟不上摄像头帧率发太快只会堆积任务。我的做法是用“上一帧处理完再发下一帧”的背压机制保证不会积压。const video document.getElementById(video); const canvas document.getElementById(overlay); const ctx canvas.getContext(2d); const worker new Worker(./worker.js, { type: module }); let busy false; async function loop() { if (!busy video.readyState 2) { busy true; const bitmap await createImageBitmap(video); worker.postMessage({ type: frame, bitmap }, [bitmap]); } requestAnimationFrame(loop); } worker.onmessage (e) { const { boxes } e.data; ctx.clearRect(0, 0, canvas.width, canvas.height); for (const box of boxes) { ctx.strokeStyle #00ff00; ctx.lineWidth 2; ctx.strokeRect(box.x, box.y, box.w, box.h); } busy false; }; loop();注意createImageBitmap是异步的而且它创建的 bitmap 是可转移对象通过postMessage的第二个参数转移给 Worker 后主线程就不再持有它避免了拷贝开销。这个小技巧能省不少内存带宽。4.3 Worker推理与后处理Worker 里初始化 ONNX Runtime 会话然后监听消息。收到帧后画到OffscreenCanvas取像素做预处理跑推理做 NMS最后把框发回去。import * as ort from onnxruntime-web; let session null; const INPUT_SIZE 640; async function init() { ort.env.wasm.wasmPaths /; session await ort.InferenceSession.create(/model.onnx, { executionProviders: [webgl, wasm], graphOptimizationLevel: all }); } self.onmessage async (e) { if (e.data.type frame) { const bitmap e.data.bitmap; const canvas new OffscreenCanvas(INPUT_SIZE, INPUT_SIZE); const ctx canvas.getContext(2d); ctx.drawImage(bitmap, 0, 0, INPUT_SIZE, INPUT_SIZE); const imageData ctx.getImageData(0, 0, INPUT_SIZE, INPUT_SIZE); const input preprocess(imageData); const tensor new ort.Tensor(float32, input, [1, 3, INPUT_SIZE, INPUT_SIZE]); const outputs await session.run({ images: tensor }); const boxes postprocess(outputs, bitmap.width, bitmap.height); self.postMessage({ boxes }); bitmap.close(); } };preprocess里做归一化和 HWC 到 CHW 的转换。postprocess里做 NMS 和坐标映射。这两个函数是性能热点建议用Float32Array和for循环避免高阶数组方法。4.4 参数计算输入尺寸与坐标映射的数学假设原始视频是 1280x720模型输入是 640x640。如果你直接拉伸宽高比就变了检测框会变形。所以要用 letterbox先算缩放比例scale min(640/1280, 640/720) 0.5然后缩放后尺寸是 640x360上下各填充(640-360)/2 140像素。后处理时模型输出的框坐标是在 640x640 空间里的。你要先减去填充量y y - 140再除以缩放比例y y / 0.5。这样得到的才是原始视频坐标系下的框。x 方向同理但因为宽度刚好是 640没有填充所以x x / 0.5。这个计算过程我建议写成独立函数并且用单元测试验证。我当初就是靠一个简单的测试用例已知输入输出才发现自己把填充量加错了方向。5. 常见问题与排查技巧实录那些文档里不会写的坑5.1 性能突然下降WebGL 后端悄悄回退了这是最隐蔽的问题之一。你明明指定了webgl但性能就是上不去。原因可能是GPU 不支持某个算子框架自动回退到了 WASM。排查方法是打开浏览器的开发者工具看控制台有没有回退警告或者在代码里监听session的后端信息。如果发现回退了要么换一个算子兼容性更好的模型要么接受 WASM 的性能。另一个性能杀手是纹理上传开销。每一帧都要把图像数据传到 GPU如果图像很大这个传输时间可能比推理本身还长。优化方法是尽量在 GPU 侧完成预处理减少 CPU 和 GPU 之间的数据往返。5.2 内存泄漏Worker 里的张量没释放ONNX Runtime Web 的Tensor对象如果频繁创建而不释放内存会持续增长。虽然它有垃圾回收但在高频推理场景下GC 跟不上分配速度。我的做法是复用输入张量的缓冲区每次只更新数据不重新创建对象。输出张量也要及时置空让 GC 能回收。还有一个容易忽略的点ImageBitmap用完必须调close()。我见过一个项目跑了几分钟后标签页直接崩溃就是因为 bitmap 没释放内存爆了。5.3 跨浏览器差异Chrome 能跑别的浏览器不行不同浏览器对 WebGL 扩展的支持不一样。比如OES_texture_float这个扩展有些浏览器支持有些不支持。如果不支持浮点纹理就用不了推理精度和速度都会受影响。我的建议是在初始化时检测关键扩展如果不支持就降级到 WASM 后端并给用户一个提示。另外Safari 对 WebGL 的实现有一些自己的“脾气”比如对纹理尺寸有限制超过一定大小会静默失败。如果你发现 Safari 上模型加载失败先检查输入尺寸是不是太大了。5.4 常见问题速查表问题现象可能原因排查方法解决思路页面卡死推理在主线程跑看 Performance 面板移到 Web Worker帧率低WebGL 回退到 WASM看控制台警告换模型或接受降级检测框偏移坐标映射错误打印中间坐标检查 letterbox 参数内存持续增长张量或 bitmap 未释放看内存快照复用缓冲区及时 close摄像头打不开非 HTTPS 环境看控制台报错上 HTTPS 或 localhost模型加载失败路径或格式错误看网络请求检查 MIME 类型和路径5.5 独家避坑技巧第一个技巧在 Worker 里做推理时把session.run包在 try-catch 里。因为 GPU 上下文可能因为各种原因丢失比如系统休眠恢复后一旦丢失后续推理全部失败。捕获到错误后重新创建 session 就能恢复。第二个技巧用performance.now()打点把预处理、推理、后处理的时间分别记录下来。这样你才知道瓶颈在哪。我一开始以为推理最慢结果发现预处理里的getImageData才是大头后来改用OffscreenCanvas的transferToImageBitmap才优化下来。第三个技巧模型输入尺寸不要盲目追求大。640 和 416 的精度差异在很多场景下肉眼几乎看不出但速度差了一倍。先用小尺寸验证链路再根据精度需求往上调。6. 端侧视觉 AI 的边界与我的实际体会把神经网络塞进浏览器标签页听起来很酷但它有明确的边界。它不适合跑超大模型不适合对延迟要求极致到毫秒级的场景也不适合需要复杂多模型串联的任务。它的甜点区是中等规模模型、实时性要求 20 FPS 以上、部署环境不可控、数据不能出端。工业质检、教育演示、隐私敏感的消费应用都是它的用武之地。我在实际项目里最大的体会是浏览器端的性能优化八成功夫在数据搬运上。模型本身的计算框架已经帮你优化得很好了。真正拖慢速度的是图像从摄像头到 CPU、从 CPU 到 GPU、从 Worker 到主线程的这些搬运过程。谁能把这些搬运做到最少、最快谁就能把帧率提上去。最后分享一个小技巧如果你的场景允许把视频帧直接以VideoFrame的形式传给 Worker而不是转成ImageBitmap。VideoFrame是更底层的对象某些浏览器上它的传输开销更小。不过兼容性需要你自己测一下不是所有浏览器都支持。这个方向后续还可以往 WebGPU 迁移等兼容性成熟后计算着色器能带来更灵活的算子实现和更好的性能。但在那之前WebGL Worker 这套组合仍然是浏览器端视觉 AI 最稳的工程方案。
