算子将输入图片从一种颜色空间转换到另一种颜色空间,支持色彩通道、格式和深度的转换,常来转换图片格式。
| 输入图像 | 参数 | 输出图像 |
|---|---|---|
![]() | - | ![]() |
颜色空间转换的过程中有多种公式可以选择,下面列举一种方法:
RGB to Gray
Y=0.299∗R+0.587∗G+0.114∗B
RGB/BGR ⇔ YUV420
Y=16+0.257∗R+0.504∗G+0.098∗B
Cb=128−0.148∗R−0.291∗G+0.439∗B
Cr=128+0.439∗R−0.368∗G−0.071∗B
R=1.164∗(Y−16)+1.596∗(Cr−128)
G=1.164∗(Y−16)−0.392∗(Cb−128)−0.812∗(Cr−128)
B=1.164∗(Y−16)+2.016∗(Cb−128)
int32_t hbVPCvtColor(hbUCPTaskHandle_t *taskHandle,
hbVPImage *dstImg,
hbVPImage const *srcImg);
详细接口信息请查看 hbVPCvtColor 。
// Include the header
#include "hobot/hb_ucp.h"
#include "hobot/vp/hb_vp.h"
#include "hobot/vp/hb_vp_cvt_color.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_RGB,
HB_VP_IMAGE_TYPE_U8C3,
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
hbVPCvtColor(&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);

