如何使用VPSS实现旋转功能

转载自：如何使用VPSS实现旋转功能
作者：ltshan139

前言

海思芯片里面VI，VPSS以及VGS都可以实现图片旋转功能。VI实现旋转的话，会导致后期保存的图片也实现了旋转；VGS实现旋转有些复杂，要多加不少代码；VPSS实现旋转相对很简单，而且不会影响原始照片。

VPSS旋转

VPSS有两个实现旋转的API：

HI_MPI_VPSS_SetChnRotation(VPSS_GRP VpssGrp, VPSS_CHN VpssChn, ROTATION_E enRotation);

HI_S32 HI_MPI_VPSS_SetChnRotationEx(VPSS_GRP VpssGrp, VPSS_CHN VpssChn, const VPSS_ROTATION_EX_ATTR_S* pstRotationExAttr);

第一个API使用非常简单，但只能实现90，180以及270度的旋转；而第二个API则能够进行任意角度的旋转。

这里只介绍第一个API的使用情况，下面代码实现90度旋转。

s32Ret = HI_MPI_VPSS_SetChnRotation(VpssGrp, VpssChn, ROTATION_90);
if(s32Ret != HI_SUCCESS)
{
    printf("HI_MPI_VPSS_SetChnRotation failed with %#x\n", s32Ret);
    return HI_FAILURE;
}

是不是非常简单？不过还是有以下几个问题需要注意的。

1）角度旋转是顺时针。比如输入参数为ROTATION_90，那就是图片顺时针旋转90度。

2）添加了旋转功能后，vb buffer数目要加1，否则HI_MPI_VPSS_GetChnFrame()会fail. 其错误log如下所示。

3）vpss是先实现scale，再做rotation。

本人vpss的channel attr 部分value如下所示

astVpssChnAttr.u32Width                    =  608;
astVpssChnAttr.u32Height                   = 416;
astVpssChnAttr.enChnMode                   = VPSS_CHN_MODE_USER;
astVpssChnAttr.enCompressMode              = COMPRESS_MODE_NONE;
astVpssChnAttr.enDynamicRange              = DYNAMIC_RANGE_SDR8;
astVpssChnAttr.enPixelFormat               = PIXEL_FORMAT_YVU_SEMIPLANAR_420; 
astVpssChnAttr.stFrameRate.s32SrcFrameRate = -1;
astVpssChnAttr.stFrameRate.s32DstFrameRate = -1;
astVpssChnAttr.u32Depth                    = 2;
astVpssChnAttr.bMirror                     = HI_FALSE;
astVpssChnAttr.bFlip                       = HI_FALSE;
astVpssChnAttr.stAspectRatio.enMode        = ASPECT_RATIO_NONE;
astVpssChnAttr.enVideoFormat               = VIDEO_FORMAT_LINEAR;

这里width和height分别设为608和416。假如原始图片为1920x1080.那么经过vpss scale后其图像变成了 608x416。这时如果vpss已经设置过90度的rotation后，那么真正从vpss获取的帧图像数据实际上就是416x608了。

举个例子，假如目标检测网络的size为608x416，如果需要对图像做90度旋转，那么vpssChnAttr的u32Width和u32Height则要设为416和608. 这样才能对应起来。

海思AI芯片系列文章

更多海思AI芯片方案学习笔记欢迎关注海思AI芯片方案学习。

前言

VPSS旋转

推荐阅读

目录