Equalizehist

直方图均衡化是一种利用图像强度分布的直方图进行对比度调整的图像处理方法。

算子效果

输入图像参数输出图像
image-image

原理

要达到均衡效果,需要对图像进行重映射,其累积分布函数计算如下:

H(i)=0j<iH(j)H'(i)=\displaystyle\sum_{0\leqslant j<i}H(j)

其中,HH 为输入直方图。

得到累积分布后需要将其规范化,得到函数形式如下:

image

最后使用重映射函数得到均衡后的图像,函数如下:

dst(x,y)=H(src(x,y))dst(x,y)=H'(src(x,y))

API接口

int32_t hbVPEqualizeHist(hbUCPTaskHandle_t *taskHandle,
                         hbVPImage *dstImg,
                         hbVPImage const *srcImg);

详细接口信息请查看 hbVPEqualizeHist

使用方法

// Include the header
#include "hobot/hb_ucp.h"
#include "hobot/vp/hb_vp.h"
#include "hobot/vp/hb_vp_equalize_hist.h"

// init Image, allocate memory for image data
hbUCPSysMem src_mem;
hbUCPMallocCached(&src_mem, src_stride * src_height, 0);
hbVPImage src_img{HB_VP_IMAGE_FORMAT_Y,
                  HB_VP_IMAGE_TYPE_U8C1,
                  src_width,
                  src_height,
                  src_stride,
                  src_mem.virAddr,
                  src_mem.phyAddr,
                  nullptr,
                  0,
                  0};

hbUCPSysMem dst_mem;
hbUCPMallocCached(&dst_mem, dst_stride * dst_height, 0);
hbVPImage dst_img{HB_VP_IMAGE_FORMAT_Y,
                  HB_VP_IMAGE_TYPE_U8C1,
                  dst_width,
                  dst_height,
                  dst_stride,
                  dst_mem.virAddr,
                  dst_mem.phyAddr,
                  nullptr,
                  0,
                  0};

// init task handle and schedule param
hbUCPTaskHandle_t task_handle{nullptr};
hbUCPSchedParam sched_param;
HB_UCP_INITIALIZE_SCHED_PARAM(&sched_param);
sched_param.backend = HB_UCP_DSP_CORE_0;

// create task
hbVPEqualizeHist(&task_handle, &dst_img, &src_img);

// 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_mem);
hbUCPFree(&dst_mem);