如何利用CANN DVPP进行图片的等比例缩放?

摘要:介绍如何用昇腾AI处理器上的DVPP单元进行,图像的等比例缩放,保证图像不变形。

本文分享自华为云社区《CANN DVPP进行图片的等比例缩放》,作者:马城林 。

1. 为什么需要进行等比例的缩放,直接暴力缩放成模型需要的宽高岂不是更省事

首先没有任何的规定表示我们必须进行等比例的缩放,因为实际上即使图像上的物体因为缩放而变形,物体本身的特征还是存在,神经网络依旧可以提取对应的特征进而预测出物体的位置,通过计算实际的宽高与模型之间的宽高的比例依旧可以将模型预测到的候选框的映射到实际的原图上。这样看来等比例的缩放存在感好低!

但是事情真的是这样吗?肯定不是呀,要不我为啥会写这篇博客呢?

首先有以下几点考虑:

  • 目标检测除了将图像送入模型,实际上物体在图像中的位置也被作为一部分信息参与了模型的训练,经过等比例缩放的图片在坐标转换上明显省事。
  • 有些图像经过非等比例压缩后已经与实际场景中的物体差别很大,严重的变形会干扰到模型的准确性。
  • 等比例缩放更加符合我们人类的直觉。

2. 如何在昇腾AI处理器上进行等比例缩放

如果在我们的训练服务器,或者我们自己的个人计算机上,因为计算能力的相对充沛,我们可以借助很多开源图像库例如opencv,pillow等进行图像的等比例缩放。但是一旦道路端侧部署时这些开源图像库的方法往往成为整个推理过程最耗时的部分,因为他们主要通过CPU进行运算,而在诸如200DK这样的环境上,进行这样的变换让本就不充沛的CPU算力雪上加霜。而昇腾AI处理器上拥有专门处理数字图像的硬件单元。所以如何借助这些专用的硬件单元是十分重要的。话不多说开始吧!

3. 实战讲解

主要从以下几个方面进行讲解

  • 步骤分解
  • 每一步在昇腾AI处理器上的实现
  • 怎么把几步组合起来完成一个完整的动作
  • 介绍 AscendCL 为开发者提供的组合API接口,只需要一个函数完成所有动作,想想都刺激

<概念注意>

  • 昇腾AI处理器在不同的环境上拥有不同运行模式,常见的分为两种,一种是类似于显卡一样插在服务器的PCIE插槽上,提供加速服务,一种是类似于200DK一样的嵌入式环境。
  • 工作在服务器的PCIE插槽上的昇腾310内存有两种存在形态,一种就是我们常见的内存,称为host内存。一种类似于显卡的显存,称为device内存,这种状态下需要涉及到内存拷贝的问题。
  • 200DK这种环境,不区分host与device,只有一块device内存,所以并不涉及host与device的内存拷贝问题
  • 媒体数据处理描述图形的大小,主要有两种表示方法,width,height,表示图像的真实宽高,widthstride,heightstride表示图像在内存中对其到128,16,2后的宽高,内存宽高标识图像在内存中的位置(但是可能有为了填充而产生的无效数据),真实宽高就标识内存中真实有效的数据。
  • JPEG Decode时对JPEG的源图像没有宽高的要求,但是输出的数据,宽会对齐到128长会对齐到16,对齐后可能会产生一部分的无效数据。
  • VPC 图像缩放,抠图,贴图对输入的源图像宽高对齐到2,对齐后的内存宽高为,宽对齐到16高对齐到2,输出与输入约束条件相同
  • JPEG Encode 对输入的源图像宽高对齐到2,对齐后的内存宽高为,宽对齐到16或者16的倍数(128时性能更好)高对齐到16

详细的约束对齐规则
JPEG Decode
VPC
JPEG Encode

3.1 步骤分解

很多朋友和我一样属于一听就会,一动手就废的类型,在脑海里勾勒出万里江山,实际一上手,我是个傻子。。。。。

实际上是我们还没有掌握做事情的诀窍,就是把事情分解成可以很方便用代码描述出的步骤;计算机编程让人感到头皮发麻的一个重要原因就是计算机它只会一步一步来,不会跳步。所以我们脑海里面很简单的步骤在编程实现后也可能代码很长。

1、将常见的JPEG图片解码成YUV格式的图片

aclError acldvppJpegDecodeAsync(acldvppChannelDesc *channelDesc, const void *data, uint32_t size, acldvppPicDesc *outputDesc,aclrtStream stream)

2、因为JPEG解码有128x16的对齐要求,所以解码后的数据边缘会出现无效数据,需要进行裁剪

aclError acldvppVpcCropAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppRoiConfig *cropArea, aclrtStream stream)

3、进行resize的操作,没有对应的等比例缩放的接口,但是如果我们最后指定的图像大小相对于原图是等比例的,就是等比例缩放

aclError acldvppVpcResizeAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppResizeConfig *resizeConfig, aclrtStream stream)

4、进行等比例缩放时需要一个,与模型大小一致的纯色背景图片作为贴图的背景,我们可以申请对应大小的一些区域然后,将对应区域全部赋值为同一个数。

aclError aclrtMemset (void *devPtr, size_t maxCount, int32_t value, size_t count)

5、进行贴图

aclError acldvppVpcCropAndPasteAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppRoiConfig *cropArea, acldvppRoiConfig *pasteArea, aclrtStream stream)

6、进行模型推理或编码保存

3.2 对应的变量怎么声明与初始化

  • DVPP处理的通道的声明与销毁

DVPP的一系列操作都是建立在已经声明初始化的通道上,你可以理解为初始化了相关的硬件资源

acldvppChannelDesc *channelDesc
channelDesc = acldvppCreateChannelDesc();
aclError ret = acldvppCreateChannel(channelDesc);
// 销毁与创建的顺序相反
acldvppDestroyChannel(channelDesc);
acldvppDestroyChannelDesc(channelDesc);
  • 各种图像的操作描述对应的图像的描述信息的创建与销毁
acldvppPicDesc *picDesc
picDesc = acldvppCreatePicDesc();
acldvppSetPicDescData(picDesc, decodeOutDevBuffer_); // 图像的内存地址
acldvppSetPicDescSize(picDesc, decodeOutBufferSize); // 图像所占的内存大小
acldvppSetPicDescFormat(picDesc, PIXEL_FORMAT_YUV_SEMIPLANAR_420); // 图像的格式
acldvppSetPicDescWidth(picDesc, inputWidth_); // 图像的实际宽
acldvppSetPicDescHeight(picDesc, inputHeight_); // 图像的实际高
acldvppSetPicDescWidthStride(picDesc, decodeOutWidthStride); // 图像在内存中对齐后的宽
acldvppSetPicDescHeightStride(picDesc, decodeOutHeightStride); // 图像在内存中对齐后的高
// 销毁
acldvppDestroyPicDesc(picDesc);
  • 申请内存
// device侧,200DK等只有device内存的也用此接口,且只用此接口
aclError aclrtMalloc(void **devPtr, size_t size, aclrtMemMallocPolicy policy)
aclError aclrtFree(void *devPtr)
// host侧
aclError aclrtMallocHost(void **hostPtr, size_t size)
aclError aclrtFreeHost(void *hostPtr)
// dvpp 内存(在device侧,因为需要内存地址128对齐,所以拥有单独的接口)
aclError acldvppMalloc(void **devPtr, size_t size)
aclError acldvppFree(void *devPtr)
  • 创建抠图,贴图的区域

如何利用CANN DVPP进行图片的等比例缩放?_第1张图片

acldvppRoiConfig *acldvppCreateRoiConfig(uint32_t left, uint32_t right, uint32_t top, uint32_t bottom)
// 销毁
aclError acldvppDestroyRoiConfig(acldvppRoiConfig *roiConfig)
  • 不知道图片的相关信息获取信息的接口
// 通过此接口可以方便的获取JPEG图像的宽,高,色彩通道数,编码格式,只需要传入对应的JPEG源数据地址以及大小
aclError acldvppJpegGetImageInfoV2(const void *data, uint32_t size, uint32_t *width, uint32_t *height, int32_t *components, acldvppJpegFormat *format)
// 获取JPEG图片解码后的数据大小,因为解码后会对齐到128,所以不能任何时候都简单地依据原图大小计算解码后占用的内存大小。
aclError acldvppJpegPredictDecSize(const void *data, uint32_t dataSize, acldvppPixelFormat outputPixelFormat, uint32_t *decSize)
// 获取PNG图像的宽和高
aclError acldvppPngGetImageInfo(const void *data, uint32_t dataSize, uint32_t *width, uint32_t *height, int32_t *components)
// 预测PNG图片解码后占用内存空间
aclError acldvppPngPredictDecSize(const void *data, uint32_t dataSize, acldvppPixelFormat outputPixelFormat, uint32_t *decSize)
// 获取JPEG图像编码后的占用内存空间
aclError acldvppJpegPredictEncSize(const acldvppPicDesc *inputDesc, const acldvppJpegeConfig *config, uint32_t *size)
  • 图像缩放,贴图,抠图相关的接口(当有多个操作动作时建议使用组合接口,虽然看似是多个步骤,实际上硬件内部只是执行一次操作可以大幅度提升处理的效率)
// 缩放接口,单一接口,将缩放后的图片作为输出
aclError acldvppVpcResizeAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppResizeConfig *resizeConfig, aclrtStream stream)
// 抠图接口,单一接口,从原图中根据ROI区域抠出一部分作为输出
aclError acldvppVpcCropAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppRoiConfig *cropArea, aclrtStream stream)
// 组合接口(抠图+缩放),应用场景:JPEG 解码后的图片中可能有无效的数据,直接缩放到模型需要的大小会把无效数据一起输入,可以先把真实图像的数据抠出来然后进行缩放,减少无效的数据干扰。
aclError acldvppVpcCropResizeAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppRoiConfig *cropArea, acldvppResizeConfig *resizeConfig, aclrtStream stream)
// 组合接口(抠图+贴图 / 抠图+缩放+贴图)如果抠图与贴图的大小不一致会进行一步缩放操作。
aclError acldvppVpcCropAndPasteAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppRoiConfig *cropArea, acldvppRoiConfig *pasteArea, aclrtStream stream)
// 组合接口(抠图+贴图 / 抠图+缩放+贴图)此接口功能与上一个是一致的,但是此接口可以指定不同的缩放算法
aclError acldvppVpcCropResizePasteAsync(acldvppChannelDesc *channelDesc, acldvppPicDesc *inputDesc, acldvppPicDesc *outputDesc, acldvppRoiConfig *cropArea, acldvppRoiConfig *pasteArea, acldvppResizeConfig *resizeConfig, aclrtStream stream)

3.3 实战讲解

效果展示:1024x688 =》 416x416

如何利用CANN DVPP进行图片的等比例缩放?_第2张图片

如何利用CANN DVPP进行图片的等比例缩放?_第3张图片

// 定义一些全局变量
namespace {
 std::string aclConfig = "../src/acl.json";
 const std::string image_path = "../data/dog2.jpg";
 const std::string image_save_path = "../out/dog2.jpg";
    int32_t deviceId = 0;
 aclrtContext context;
 aclrtStream stream;
 aclrtRunMode runMode_ = ACL_DEVICE;
    uint32_t model_width = 416; // 模型需要的图像宽
    uint32_t model_height = 416; // 模型需要的图像高
    uint32_t image_width = 0; // 图像的真实宽
    uint32_t image_height = 0; // 图像的真实高
 acldvppChannelDesc *dvppChannelDesc = nullptr;
}

开始进行解码操作

// 将图片读入内存,如果运行模式为(ACL_HOST)模式,需要申请host侧内存存放我们的原始图片数据,然后申请相同大小的device内存拷贝过去。如果运行在(ACL_DEVICE)模式,例如200DK这样的场景,直接申请device内存,不需要进行内存的拷贝搬运。
// 经过读取图片的步骤我们应该能获得如下变量
void* JpegImagePtr; // 原始图像在device侧的内存地址(不管运行在那种方式下的昇腾软件栈都是如此)
uint32_t JpegImageSize; // 原始图像的大小
// 我们不了解图像的编码以及真实宽高,还有颜色通道数,所以需要执行获取图像信息的接口
// 图像的原始宽高变量我们前面已经定义过, 所以我们现在只需要定义图像的通道数与编码方式还有图像解码后所占用的内存的大小
int32_t components; // 图像的通道数
acldvppJpegFormat format; // 图像的编码方式
uint32_t JpegdOutSize;
void* JpegdOutPtr;
aclError ret = acldvppJpegGetImageInfoV2(JpegImagePtr, JpegImageSize, &image_width, &height, &components, &format)
aclError ret = acldvppJpegPredictDecSize(JpegImagePtr, JpegImageSize, format, &JpegdOutSize)
// 申请存放解码后YUV图像的内存
aclError ret = acldvppMalloc(&JpegdOutPtr, JpegdOutSize);
// 设置JPEG解码后图像的描述信息(注意JPEG解码后内存宽高对齐到128x16)
acldvppPicDesc* JpegdOutputPicDesc = acldvppCreatePicDesc();
acldvppSetPicDescData(out_pic_desc, JpegdOutPtr);
acldvppSetPicDescSize(out_pic_desc, JpegdOutSize);
acldvppSetPicDescFormat(out_pic_desc, PIXEL_FORMAT_YUV_SEMIPLANAR_420);
acldvppSetPicDescWidth(out_pic_desc, image_width);
acldvppSetPicDescHeight(out_pic_desc, image_height);
acldvppSetPicDescWidthStride(out_pic_desc, ALIGN_UP128(image_width));
acldvppSetPicDescHeightStride(out_pic_desc, ALIGN_UP16(image_height));
// 执行解码接口
ret = acldvppJpegDecodeAsync(dvppChannelDesc, JpegImagePtr, JpegImageSize, JpegdOutputPicDesc, stream);
ret = aclrtSynchronizeStream(stream_);

开始进行抠图的操作,因为我们这次需要执行抠图+缩放+贴图三个动作所以我们选择组合API进行编程

// 首先声明原图抠图的区域
// 抠图的区域左偏移和上偏移都是偶数
// 抠图的区域右偏移和下偏移都是奇数
uint32_t left = 0;
uint32_t top = 0;
uint32_t right = image_width % 2 == 0 ? image_width-1 : image_width;
uint32_t bottom = image_height % 2 == 0 ? image_height-1 : image_height;
acldvppRoiConfig* cropArea = cropConfig = acldvppCreateRoiConfig(left, right, top, bottom);
// 计算等比例缩放后的新的宽高(等比例缩放后可能不满足16x2的对齐规则,需要手动进行对齐的操作)
void LargeSizeAtLeast(uint32_t W, uint32_t H, uint32_t &newInputWidth, uint32_t &newInputHeight)
{
 INFO_LOG("W:%u H:%u nw:%u, nh:%u", W, H, newInputWidth, newInputHeight);
    float scaleRatio = 0.0;
    float inputWidth = 0.0;
    float inputHeight = 0.0;
    float resizeMax = 0.0;
    bool maxWidthFlag = false;
 inputWidth = (float)W;
 inputHeight = (float)H;
 resizeMax = (float)(416);
 maxWidthFlag = (W >= H) ? true : false;
 if (maxWidthFlag == true) {
 newInputWidth = resizeMax;
 scaleRatio = resizeMax / W;
 // 高度2对齐
 newInputHeight = scaleRatio * H;
 newInputHeight = ALIGN_UP2(newInputHeight);
 INFO_LOG("scaleRatio: %.3f, modelInputWidth: %u, modelInputHeight: %d, newInputWidth: %d, newInputHeight: %d",
 scaleRatio, W, H, newInputWidth, newInputHeight);
 } else {
 scaleRatio = resizeMax / H;
 // 如果高度是长边,建议宽度在等比例缩放后再做一次16对齐。因为vpc在输出时宽有16字节对齐约束,当贴图的宽非16对齐时,会导致在贴图的时候,
 // 芯片会自动进行16字节对齐,导致每次写入数据的时候都会引入部分无效数据,从而导致精度下降。
 newInputWidth = scaleRatio * W;
 newInputWidth = ALIGN_UP16(newInputWidth);
 newInputHeight = resizeMax;
 INFO_LOG("scaleRatio: %.3f, modelInputWidth: %u, modelInputHeight: %d, newInputWidth: %d, newInputHeight: %d",
 scaleRatio, W, H, newInputWidth, newInputHeight);
 }
}
// 设置贴图的范围左偏移要求16对齐
acldvppRoiConfig *InitVpcOutConfig(uint32_t width, uint32_t height, uint32_t modelInputWidth, uint32_t modelInputHeight)
{
    uint32_t right = 0;
    uint32_t bottom = 0;
    uint32_t left = 0;
    uint32_t top = 0;
    uint32_t left_stride;
 acldvppRoiConfig *cropConfig;
    uint32_t small = width < height ? width : height;
    uint32_t padded_size_half;
 if (small == width) {
 padded_size_half = (modelInputWidth - width) / 2; // 贴图区域距离左边界的距离
        left = padded_size_half;
 left_stride = ALIGN_UP16(left);
        right = (left_stride + width) % 2 == 0 ? (left_stride + width - 1) : (left_stride + width);
 if (left_stride + right > modelInputWidth) {
 while (true) {
 left_stride = left_stride - 16;
                right = (left_stride + width) % 2 == 0 ? (left_stride + width - 1) : (left_stride + width);
 if (left_stride + right < modelInputWidth)
 break;
 }
 }
        right = (left_stride + width) % 2 == 0 ? (left_stride + width - 1) : (left_stride + width);
        bottom = (modelInputHeight % 2 == 0 ? modelInputHeight - 1 : modelInputHeight);
        top = bottom - height + 1;
 } else {
 padded_size_half = (modelInputHeight - height) / 2;
        right = (modelInputWidth % 2 == 0 ? modelInputWidth - 1 : modelInputWidth);
        left = right + 1 - width;
 left_stride = ALIGN_UP16(left);
        top = (padded_size_half % 2 == 0 ? padded_size_half : padded_size_half + 1);
        bottom = (height + top - 1) % 2 == 0 ? (height + top - 2) : (height + top - 1);
 }
 INFO_LOG("left_stride=%d, right=%d, top=%d, bottom=%d\n", left_stride, right, top, bottom);
 cropConfig = acldvppCreateRoiConfig(left_stride, right, top, bottom);
 if (cropConfig == nullptr) {
 ERROR_LOG("acldvppCreateRoiConfig failed");
 return nullptr;
 }
 return cropConfig;
}
// 设置最后完成等比例缩放后的图像的信息
void* output_ptr = nullptr;
uint32_t output_size = YUV420SP_SIZE(ALIGN_UP16(model_width), ALIGN_UP2(model_height));
ret = acldvppMalloc(&output_ptr, output_size); // 申请模板内存
if (ret != ACL_SUCCESS) {
ERROR_LOG("acl malloc output is failed");
}
ret = aclrtMemset(output_ptr, output_size, 128, output_size);
if (ret != ACL_SUCCESS) {
ERROR_LOG("mem set 128 is failed");
}
acldvppPicDesc *output_Desc = acldvppCreatePicDesc();
acldvppSetPicDescData(output_Desc, output_ptr);
acldvppSetPicDescSize(output_Desc, output_size);
acldvppSetPicDescFormat(output_Desc, PIXEL_FORMAT_YUV_SEMIPLANAR_420);
acldvppSetPicDescWidth(output_Desc, model_width);
acldvppSetPicDescHeight(output_Desc, model_height);
acldvppSetPicDescWidthStride(output_Desc, ALIGN_UP16(model_width));
acldvppSetPicDescHeightStride(output_Desc, ALIGN_UP2(model_height));
// 执行裁剪加贴图的操作
ret = acldvppVpcCropAndPasteAsync(dvppChannelDesc, JpegdOutputPicDesc, output_Desc, cropArea, pasteArea, stream);

完成了一系列的操作

总结

  • 对于DVPP处理来说,单独的接口调用需要全部满足输入和输出的长宽以及对齐的要求
  • 但是对于 JPEG+VPC 这样的串联编程来说前一个组件的输出图像描述是后一个组件的输入描述,设定对齐内存是为了有效标定,图像在内存中的位置,如果固执的设定对齐后的值,对于后一个组件来说很有可能得到的输入是有缺失的信息。
  • 根据众多对齐要求我们可以看出,对于图像的原始宽高全部2对齐是一个比较好的习惯,而且常用的图像规格都是偶数例如1920x1080,1280x720, 640x640, 416x416,最好不要出现长或者宽为奇数的情况,因为硬件会自动为奇数进行对齐从而产生无效数据。增加了处理的难度。

点击关注,第一时间了解华为云新鲜技术~

你可能感兴趣的:(技术交流,计算机视觉,人工智能,深度学习,昇腾AI处理器,图像处理)