最近编码涉及到OpenCV Mat的三维数组,特记录一下其内存存储方式。
三维Mat有多种构造方式,以下是其中一种,
int shape[]{ 5, 5, 10};
Mat td_arr(3, shape, CV_32SC1);
以该数组为例,是一个5x5x10的三维数组,数据类型为int,其在内存中的存储顺序为 (0,0,0)->(0,0,9)=>(0,1,0)->(0,1,9)=>…=>(1,0,0)->(1,0,9)=>…=>(4,4,0)->(4,4,9)
数据可通过索引(r,c,d)进行at访问,即
int val = td_arr.at<int>(r,c,d);
也可根据step[0],step[1],step[2]计算偏移量然后进行访问,
int offset = r * td_arr.step[0] + c * td_arr.step[1] + d * td_arr.step[2];
val = *((int*)(td_arr.data + offset))
第二种方式可用于对第三维数据的连续访问,假设需要获取第r行c列的D个数据进行操作,具体方式如下:
int offset = r * td_arr.step[0] + c * td_arr.step[1];
//获取第三维数据的头指针,内存连续的数据
int *head = (int*)(td_arr.data + offset);
//对连续数据进行操作,如memcpy,并行加速等
注:只有最后一维的内存连续,其他内存不一定连续,可能存在stride,但是使用step能够计算出正确的偏移量。在内存连续的情况下,假设数据类型为dtype,
step[0] = c *d *sizeof(dtype)
step[1] = d*sizeof(dtype)
step[2] = sizeof(dtype)
上述内存访问也可推广到高维数组。粗浅理解,如有误请批评指正。