bmcv_image_crop

该接口实现从一幅原图中 crop 出若干个小图。

处理器型号支持:

该接口支持BM1684/BM1684X。

接口形式:

bm_status_t bmcv_image_crop(
        bm_handle_t         handle,
        int                 crop_num,
        bmcv_rect_t*        rects,
        bm_image            input,
        bm_image*           output
);

参数说明:

  • bm_handle_t handle

    输入参数。 bm_handle 句柄。

  • int crop_num

    输入参数。需要 crop 小图的数量,既是指针 rects 所指向内容的长度,也是输出 bm_image 的数量。

  • bmcv_rect_t* rects

    输入参数。表示crop相关的信息,包括起始坐标、crop宽高等,具体内容参考下边的数据类型说明。该指针指向了若干个 crop 框的信息,框的个数由 crop_num 决定。

  • bm_image input

    输入参数。输入的 bm_image,bm_image 需要外部调用 bmcv_image _create 创建。image 内存可以使用 bm_image_alloc_dev_mem 或者 bm_image_copy_host_to_device 来开辟新的内存,或者使用 bmcv_image_attach 来 attach 已有的内存。

  • bm_image* output

    输出参数。输出 bm_image 的指针,其数量即为 crop_num。bm_image 需要外部调用 bmcv_image_create 创建。image 内存可以通过 bm_image_alloc_dev_mem 来开辟新的内存,或者使用 bmcv_image_attach 来 attach 已有的内存。如果不主动分配将在 api 内部进行自行分配。

返回值说明:

  • BM_SUCCESS: 成功

  • 其他:失败

数据类型说明:

typedef struct bmcv_rect {
    int start_x;
    int start_y;
    int crop_w;
    int crop_h;
} bmcv_rect_t;
  • start_x 描述了 crop 图像在原图中所在的起始横坐标。自左而右从 0 开始,取值范围 [0, width)。

  • start_y 描述了 crop 图像在原图中所在的起始纵坐标。自上而下从 0 开始,取值范围 [0, height)。

  • crop_w 描述的 crop 图像的宽度,也就是对应输出图像的宽度。

  • crop_h 描述的 crop 图像的高度,也就是对应输出图像的高度。

格式支持:

crop 目前支持以下 image_format:

num

image_format

1

FORMAT_BGR_PACKED

2

FORMAT_BGR_PLANAR

3

FORMAT_RGB_PACKED

4

FORMAT_RGB_PLANAR

5

FORMAT_GRAY

bm1684 crop 目前支持以下 data_type:

num

data_type

1

DATA_TYPE_EXT_FLOAT32

2

DATA_TYPE_EXT_1N_BYTE

3

DATA_TYPE_EXT_1N_BYTE_SIGNED

bm1684x crop 目前支持以下data_type:

num

data_type

1

DATA_TYPE_EXT_1N_BYTE

注意事项:

1、在调用 bmcv_image_crop()之前必须确保输入的 image 内存已经申请。

2、input output 的 data_type,image_format必须相同。

3、为了避免内存越界, start_x + crop_w 必须小于等于输入图像的 width, start_y + crop_h 必须小于等于输入图像的 height。

代码示例:

int channel   = 3;
int in_w      = 400;
int in_h      = 400;
int out_w     = 800;
int out_h     = 800;
int    dev_id = 0;
bm_handle_t handle;
bm_status_t dev_ret = bm_dev_request(&handle, dev_id);
std::shared_ptr<unsigned char> src_ptr(
        new unsigned char[channel * in_w * in_h],
        std::default_delete<unsigned char[]>());
std::shared_ptr<unsigned char> res_ptr(
        new unsigned char[channel * out_w * out_h],
        std::default_delete<unsigned char[]>());
unsigned char * src_data = src_ptr.get();
unsigned char * res_data = res_ptr.get();
for (int i = 0; i < channel * in_w * in_h; i++) {
    src_data[i] = rand() % 255;
}
// calculate res
bmcv_rect_t crop_attr;
crop_attr.start_x   = 0;
crop_attr.start_y   = 0;
crop_attr.crop_w    = 50;
crop_attr.crop_h    = 50;
bm_image input, output;
bm_image_create(handle,
        in_h,
        in_w,
        FORMAT_RGB_PLANAR,
        DATA_TYPE_EXT_1N_BYTE,
        &input);
bm_image_alloc_dev_mem(input);
bm_image_copy_host_to_device(input, (void **)&src_data);
bm_image_create(handle,
        out_h,
        out_w,
        FORMAT_RGB_PLANAR,
        DATA_TYPE_EXT_1N_BYTE,
        &output);
bm_image_alloc_dev_mem(output);
if (BM_SUCCESS != bmcv_image_crop(handle, 1, &crop_attr, input, &output)) {
    std::cout << "bmcv_copy_to error !!!" << std::endl;
    bm_image_destroy(input);
    bm_image_destroy(output);
    bm_dev_free(handle);
    exit(-1);
}
bm_image_copy_device_to_host(output, (void **)&res_data);
bm_image_destroy(input);
bm_image_destroy(output);
bm_dev_free(handle);