bmcv_image_lkpyramid¶

LK金字塔光流算法。完整的使用步骤包括创建、执行、销毁三步。该算法前半部分使用TPU，而后半部分为串行运算需要使用CPU，因此对于PCIe模式，建议使能CPU进行加速，具体步骤参考第5章节。

创建¶

由于该算法的内部实现需要一些缓存空间，为了避免重复申请释放空间，将一些准备工作封装在该创建接口中，只需要在启动前调用一次便可以多次调用execute接口（创建函数参数不变的情况下），接口形式如下：

bm_status_t bmcv_image_lkpyramid_create_plan(
        bm_handle_t handle,
        void*& plan,
        int width,
        int height,
        int winW = 21,
        int winH = 21,
        int maxLevel = 3);

输入参数说明：

bm_handle_t handle

输入参数。bm_handle 句柄

void*& plan

输出参数。执行阶段所需要的句柄。

int width

输入参数。待处理图像的宽度。

int height

输入参数，待处理图像的高度。

int winW

输入参数，算法处理窗口的宽度，默认值为21。

int winH

输入参数，算法处理窗口的高度，默认值为21。

int maxLevel

输入参数，金字塔处理的高度，默认值为3, 目前支持的最大值为5。该参数值越大，算法执行时间越长，建议根据实际效果选择可接受的最小值。

返回值说明:

BM_SUCCESS: 成功
其他：失败

执行¶

使用上述接口创建后的plan就可以开始真正的执行阶段了，接口格式如下：

typedef struct {
    float x;
    float y;
} bmcv_point2f_t;

typedef struct {
    int type;   // 1: maxCount   2: eps   3: both
    int max_count;
    double epsilon;
} bmcv_term_criteria_t;

bm_status_t bmcv_image_lkpyramid_execute(
        bm_handle_t handle,
        void* plan,
        bm_image prevImg,
        bm_image nextImg,
        int ptsNum,
        bmcv_point2f_t* prevPts,
        bmcv_point2f_t* nextPts,
        bool* status,
        bmcv_term_criteria_t criteria = {3, 30, 0.01});

输入参数说明：

bm_handle_t handle

输入参数。bm_handle 句柄

const void *plan

输入参数。创建阶段所得到的句柄。

bm_image prevImg

输入参数。前一幅图像的 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 nextImg

输入参数。后一幅图像的 bm_image，bm_image 需要外部调用 bmcv_image_create 创建。image 内存可以使用 bm_image_alloc_dev_mem 或者 bm_image_copy_host_to_device 来开辟新的内存，或者使用 bmcv_image_attach 来 attach 已有的内存。

int ptsNum

输入参数。需要追踪点的数量。

bmcv_point2f_t* prevPts

输入参数。需要追踪点在前一幅图中的坐标指针，其指向的长度为ptsNum。

bmcv_point2f_t* nextPts

输出参数。计算得到的追踪点在后一张图像中坐标指针，其指向的长度为ptsNum。

bool* status

输出参数。nextPts中的各个追踪点是否有效，其指向的长度为ptsNum，与nextPts中的坐标一一对应，如果有效则为true，否则为false（表示没有在后一张图像中找到对应的跟踪点，可能超出图像范围）。

bmcv_term_criteria_t criteria

输入参数。迭代结束标准，type表示以哪个参数作为结束判断条件：若为1则以迭代次数max_count为结束判断参数，若为2则以误差epsilon为结束判断参数，若为3则两者均需满足。该参数会影响执行时间，建议根据实际效果选择最优的停止迭代标准。

返回值说明:

BM_SUCCESS: 成功
其他:失败

销毁¶

当执行完成后需要销毁所创建的句柄。该接口必须和创建接口bmcv_image_lkpyramid_create_plan成对使用。

void bmcv_image_lkpyramid_destroy_plan(bm_handle_t handle, void *plan);

格式支持：

该接口目前支持以下 image_format:

num	image_format
1	FORMAT_GRAY

目前支持以下 data_type:

num	data_type
1	DATA_TYPE_EXT_1N_BYTE

示例代码¶

bm_handle_t handle;
bm_status_t ret = bm_dev_request(&handle, 0);
if (ret != BM_SUCCESS) {
    printf("Create bm handle failed. ret = %d\n", ret);
    return -1;
}
ret = bmcv_open_cpu_process(handle);
if (ret != BM_SUCCESS) {
    printf("BMCV enable CPU failed. ret = %d\n", ret);
    bm_dev_free(handle);
    return -1;
}
bm_image_format_ext fmt = FORMAT_GRAY;
bm_image prevImg;
bm_image nextImg;
bm_image_create(handle, height, width, fmt, DATA_TYPE_EXT_1N_BYTE, &prevImg);
bm_image_create(handle, height, width, fmt, DATA_TYPE_EXT_1N_BYTE, &nextImg);
bm_image_alloc_dev_mem(prevImg);
bm_image_alloc_dev_mem(nextImg);
bm_image_copy_host_to_device(prevImg, (void **)(&prevPtr));
bm_image_copy_host_to_device(nextImg, (void **)(&nextPtr));
void *plan = nullptr;
bmcv_image_lkpyramid_create_plan(
        handle,
        plan,
        width,
        height,
        kw,
        kh,
        maxLevel);
bmcv_image_lkpyramid_execute(
        handle,
        plan,
        prevImg,
        nextImg,
        ptsNum,
        prevPts,
        nextPts,
        status,
        criteria);
bmcv_image_lkpyramid_destroy_plan(handle, plan);
bm_image_destroy(prevImg);
bm_image_destroy(nextImg);
ret = bmcv_close_cpu_process(handle);
if (ret != BM_SUCCESS) {
    printf("BMCV disable CPU failed. ret = %d\n", ret);
    bm_dev_free(handle);
    return -1;
}
bm_dev_free(handle);