EdgeBoard嵌入式AI解决方案中NHWC数据格式的性能分析

时间:2025-05-03  作者:Diven  阅读:0

EdgeBoard简介

EdgeBoard嵌入式AI解决方案中NHWC数据格式的性能分析

EdgeBoard是百度基于FPGA芯片研发的嵌入式AI解决方案,高性能的加速引擎可提供3.6Tops的强大算力,完整的嵌入式参考设计使硬件集成轻松便捷。目前EdgeBoard提供了FPGA软核和计算卡模块两种形态供硬件集成,面向项目部署也提供了抓拍机和计算盒两种基础硬件产品。EdgeBoard深度兼容百度大脑模型资源与工具平台(EasyDL/AI Studio),极大降低了开发验证、产品集成、科研教学、项目部署门槛,适用于安防监控、工业质检、医疗诊断、农作物生长监控、无人驾驶、无人零售等场景

数据格式的逻辑表示与物理表示

深度学习中经常会使用NCHW、NHWC和CHWN数据格式来表示数据,其中N、H、W、C定义如下:

N:一个batch内图片的数量,一次处理的图片数量

H:垂直高度方向的像素个数,图片的高

W:水平宽度方向的像素个数,图片的宽

C:通道数。例如灰度图像为1, 彩色RGB图像为3

下图表示N=2,C=16,H=5,W=4的数据排列,其中左图是逻辑表示,右图是物理表示。

以NCHW为例,其逻辑表示如左上图,n=0时,三个坐标分别标识了C、H、W的方向。第一个元素是000,第二个元素沿着w方向,即001,随后是002, 003;然后沿H方向,即004, 005, 006, 007.。.如此反复到019后;再沿C方向,020,021, 022 。..。.一直到319;再沿N方向,也就是n=1,然后重复W方向,H方向和C方向。

根据以上NCHW的划分,物理地址表示定义如下(如右上图):

[a:0] 表示W方向,在一行内从左到右

[a:1] 表示从H方向,一行一行的从上到下

[a:2] 表示在C方向,从一个通道到另外一个通道

[a:3] 表示从N方向,从n=0 到n=1

最终NCHW数据格式的物理分布(在内存中的一维表示)表示为000 001 002 003 004 。.. 018 019 020 。.. 318 319 320 。.. 。.. 637 638 639。可以理解成把一个通道的所有像素一行一行地排列起来,然后排下一个通道,即n=0排列完后再排n=1。

同理NHWC表示是先沿C方向,再沿W方向,再沿H方向,最后沿N方向。所以在内存的存放顺序是,第一个元素是000,第二个沿C方向,即020,040, 060 。..一直到300,之后切换到W方向,001 021 041 061.。.301.。到了303后再切换到H方向,即004, 024 。.. 304,最后到了319,再切换到N方向,320,340 。..一直到639。

[b:0] 表示C方向,第一个像素从一个通道到另外一个通道

[b:1] 表示从W方向,最后一个通道第一个像素回到第一个通道的第二个像素

[b:2] 表示在H方向,最后一个通道第一行最后一个像素回到第一个通道的第二行的第一个像素

[b:3] 表示从N方向,从n=0 到n=1

NHWC其物理表示为000 020 。.. 300 001 021 。.. 283 303 004 。.. 319 320 340 。.. 。.. 339 359 。.. 639。可以理解成把一个batch的一个像素的所有通道先排列起来,然后排下一个像素。n=0排列完成后,再排n=1。

同理CHWN其逻辑表示,先沿N方向,再沿W方向,再沿H方向,最后是沿C方向。

[c:0] 表示从N方向,从n=0的第一个像素到n=1的第一个像素

[c:1] 表示从N方向,从n=1的第一个像素回到n=0的第二个像素

[c:2] 表示在H方向,从n=1的第一行最后一个像素回到n=0的第二行第一个像素

[c:3] 表示从N方向,从n=1的第一个通道最后一个像素回到n=0的第二个通道第一个像素

CHWN其物理表示为 000 032 001 321 。.. 003 323 004 324 。.. 。.. 019 339 020 。..。可以理解成先把一个batch中N幅图像的第一个通道第一个像素排列起来,然后排第二个像素;再排第二个通道,第三个通道。..

数据在内存中的偏移地址

深度学习中涉及大量的数据计算,计算需要从内存中取出数据,因此需要计算出数据的偏移地址以便进行取数。有了上面的逻辑表示和物理表示,可以推导出4维逻辑表示(n,c,h,w)映射到一维内存中偏移地址的公式。

定义位置(n,c,h,w)表示第n个batch的第c通道的第h行的第w列,那么该位置在不同数据格式下,在内存中的偏移地址计算公式如下: NCHW: offset_nchw(n, c, h, w) = n * CHW + c * HW + h * W + w NHWC: offset_nhwc(n, c, h, w) = n * HWC + h * WC + w * C + c CHWN: offset_chwn(n, c, h, w) = c * HWN + h * WN + w * N + n 其中N、C、H、W为常量,n、c、h、w为变量

在NCHW中,CHW=C*H*W,表示一个Batch,可以理解成一个BGR 3通道的图片,表达的是一个立方体。HW=H*W,表示一个平面,可以理解成是BGR 3通道图片的一个通道(灰度图就是一个通道图片)。W是一行,可以理解成一个通道上的一行。

以上图为例,如果想计算出绿色圈,即341的位置(n=1,c=1, h=0, w=1)。我们需要先跳过n=0的数据(CHW),图中箭头1指向的蓝色框区域;再跳过n=1的第一个通道(HW),图中箭头2指向蓝色框区域;这时进入到了n=1的第二个通道,跳过h=0行(0*W);最后跳过w个数到达偏移位置。

EdgeBoard为何使用NHWC

下面来分析EdgeBoard选择NHWC数据格式的原因。

上图表示卷积的计算过程。根据卷积的运算特点,相同位置窗口所有通道数与卷积的参数相乘后累加,可以有下面两种计算方式:

先像素后通道:先把一个通道滑动窗口与卷积参数相乘后累加,再进行下一个通道,直到所有通道乘累加完成。比如第一次滑窗计算公式:

可以看出两种方式计算的结果是一样。

而对于NHWC格式,即先通道后像素,是把一个像素的所有通道的数据放在一起。这样对应上图第一个像素的3个通道值,第二个像素的3个通道值,第三个像素的3个通道值,在内存中的地址都是连续的,也就是说一次就可以把kernel第一行需要计算的数取出,3x3的kernel需要3次取数。

而对于NCHW格式,即先像素后通道,是把一个通道的所有像素按顺序排列,这样对于一个3*3的卷积核,需要每取3个数就需要跳跃n个数后,再取3个数。一个通道需要取3次,3个通道需要取9次。

在实际网络中,通常通道数会远大于卷积kernel数(不会像上图只有3个通道,通常是几十、几百个通道)。这样对于NHWC格式来说说,取数的次数会比NCHW少很多。对EdgeBoard来说,为了增加其所支持网络的性,减少对大输入尺寸和高存储量权重网络的限制,采用NHWC的格式可以实现分批次地把Feature Map和Weight数据读取到FPGA的片上缓存,例如对于3x3的Kernel,我们可以只读取三行(3WC)Feature Map的数据到FPGA内进行计算,即可得到一行输出数据,并传输到片外大容量缓存DDR,而不需依赖下一个3WC的Feature Map输入数据就可完成每一批次的输入输出数据传输。

再例如,我们也可将Weight数据根据FPGA片上缓存的不同大小分割成N份,一份一份发送到FPGA分别做卷积运算后,再传输回DDR做相应拼接,这等同于做一次大的卷积运算,好处在于可以根据不同容量的FPGA器件做不同匹配,大大提高了代码的硬件适配性。由于C维度之间数据相关性较弱,采用NHWC格式更能有着FPGA高并行度的计算特点,充分利用FPGA的算力。

下表为EdgeBoard使用NHWC数据格式的网络性能:

猜您喜欢

吊牌和标签在产品展示中是重要配件,不仅传递信息,还能提升品牌形象。了解吊牌和标签的参数,有助于更好地设计和使用。材质是吊牌和标签的重要参数。常见的材质包括纸质、...
2020-01-13 00:00:00

贴片电阻体积小巧,其阻值、误差等信息通常以数字或色环标注在表面。理解这些标记对于正确使用至关重要。数字标识: 三位数字标识法最为常见。前两位数字表示有效数字,第...
2024-11-26 11:30:03

现代科技发展的浪潮中,各种配件的应用越来越。尤其是“Accessories_6X7.6MM_TM”这一款配件,因其独特的规格和功能,受到了众多消费者的青睐。本文...
2025-04-22 14:31:47

现代科技迅速发展的时代,配件成为了提升设备性能的重要组成部分。“Accessories_35X6.35MM_TM”是一种广泛应用于多个领域的配件,特别是在机械和...
2025-02-27 00:03:27

事实上,MCU对有些任务来说是很适合的,但对其它一些任务来说可能做的并不好。举例来说,当需要并行执行大量计算任务时,FPGA可能会让你喜出望外。如果我们进一步...
2019-07-03 14:52:00

现代电子产品中,电源管理起着很重要的作用,而八个脚开关电源芯片凭借其高效、紧凑和灵活的设计,成为众多电子设备的理想选择。这种电源芯片不仅结构简单,而且具有极高的...
2024-07-23 00:00:00

太阳诱电(TAIYO YUDEN)作为一家知名的制造商,高品质的产品赢得了赞誉。其中,四端子电阻作为重要的电子元件,被应用于各种电路中。本文将深入探讨太阳诱电四...
2014-01-31 12:42:30

贴片电阻上的「104」并非直接表示电阻值的大小,而是一种编码方式。它代表的是10乘以10的4次方,单位是欧姆 (Ω)。具体计算如下:10 * 10^4 Ω = ...
2024-11-26 11:29:54

现代工业和消费市场中,配件的选择与应用越来越受到重视。特别是那些小型配件,如“Accessories_D3.9X7.8MM”,在产品性能和用户体验中是重要配件。...
2025-04-25 01:30:50

接杆作为连接电缆和设备的重要组件,其参数直接影响到电力系统的稳定性和安全性。接杆的材质是一个关键因素,常见的有铝合金和铜材,前者轻便且具有良好的导电性,而后者则...
2012-08-29 00:00:00