如何使用VPSS实现旋转功能

前言

海思芯片里面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)