OpenExplorer
地平线官网
地平线智能汽车开发者社区
地平线官网
地平线智能汽车开发者社区
中文

Inverse Fast Fourier Transform 2D

二维快速傅里叶逆变换（Inverse Fast Fourier Transform 2D，以下简称IFFT2D）是基于快速傅里叶逆变换（以下简称IFFT）开发的快速算法， 实现了二维数据从频域到时域的快速转换，通常被用来将图像等二维频域数据复原为时域数据。

算子效果

时域输入数据参数频域输出数据
imagep_size = HB_HPL_FFT16
normalize = 0
dataType = HB_HPL_DATA_TYPE_I16
imFormat = HB_IM_FORMAT_SEPARATE
numDimensionSize = 2		image

原理

IFFT2D是基于IFFT算法开发的扩展方法，对二维数据分别从y方向和x方向执行IFFT操作，实现频域数据到时域数据的映射。具体流程如下：

  1. 对二维数据中每一列进行IFFT计算，其中ny为接口参数中设置的y轴FFT点数。
HPL5IFFT2D_y
  1. 在步骤1的基础上，对二维数据中每一行进行IFFT计算，其中nx为接口参数中设置的x轴FFT点数。
HPL5IFFT2D_x
  1. 返回步骤2的计算结果。

API接口

int32_t hbIFFT2D(hbUCPTaskHandle_t *taskHandle, hbHPLImaginaryData *dst, hbHPLImaginaryData const *src, hbIFFT2DParam const *param);

详细接口信息请查看 hbIFFT2D。

使用方法

// Include the header #include "hobot/hb_ucp.h" #include "hobot/hpl/hb_hpl.h" #include "hobot/hpl/hb_ifft_2d.h" // init, allocate memory for data src_length = 1024 * 1024 * 5; hbUCPMallocCached(&src_re_mem, src_length, 0); hbUCPMallocCached(&src_im_mem, src_length, 0); hbHPLImaginaryData src; src.realDataVirAddr = src_re_mem.virAddr; src.realDataPhyAddr = src_re_mem.phyAddr; src.imDataVirAddr = src_im_mem.virAddr; src.imDataPhyAddr = src_im_mem.phyAddr; src.numDimensionSize = 2; src.dataType = HB_HPL_DATA_TYPE_I16; src.imFormat = HB_IM_FORMAT_SEPARATE; src.dimensionSize[0] = 16 * 11; src.dimensionSize[1] = 32 * 2; hbUCPMallocCached(&dst_re_mem, src_length, 0); hbUCPMallocCached(&dst_im_mem, src_length, 0); hbHPLImaginaryData dst; dst.realDataVirAddr = dst_re_mem.virAddr; dst.realDataPhyAddr = dst_re_mem.phyAddr; dst.imDataVirAddr = dst_im_mem.virAddr; dst.imDataPhyAddr = dst_im_mem.phyAddr; dst.numDimensionSize = 2; dst.dataType = HB_HPL_DATA_TYPE_I16; dst.imFormat = HB_IM_FORMAT_SEPARATE; dst.dimensionSize[0] = 16 * 11; dst.dimensionSize[1] = 32 * 2; // init param hbIFFT2DParam param; param.nx = HB_FFT_POINT_SIZE_16; param.ny = HB_FFT_POINT_SIZE_32; param.normalize = 0; // init task handle and schedule param hbUCPTaskHandle_t task_handle{nullptr}; hbUCPSchedParam sched_param{...}; // create task hbIFFT2D(&task_handle, &dst, &src, &param); // submit task hbUCPSubmitTask(task_handle, &sched_param); // wait for task done hbUCPWaitTaskDone(task_handle, 0); // release task handle hbUCPReleaseTask(task_handle); // release memory hbUCPFree(&src_re_mem); hbUCPFree(&src_im_mem); hbUCPFree(&dst_re_mem); hbUCPFree(&dst_im_mem);