bmcv_image_laplacian
梯度计算laplacian算子。
接口形式:
bm_status_t bmcv_image_laplacian( bm_handle_t handle, bm_image input, bm_image output, unsigned int ksize);
参数说明:
bm_handle_t handle
输入参数。 bm_handle 句柄。
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,bm_image 需要外部调用 bmcv_image_create 创建。image 内存可以通过 bm_image_alloc_dev_mem 来开辟新的内存,或者使用 bmcv_image_attach 来 attach 已有的内存。如果不主动分配将在 api 内部进行自行分配。
int ksize = 3
Laplacian核的大小,必须是1或3。
返回值说明:
BM_SUCCESS: 成功
其他:失败
格式支持:
该接口目前支持以下 image_format:
num |
input image_format |
output image_format |
---|---|---|
1 |
FORMAT_GRAY |
FORMAT_GRAY |
目前支持以下 data_type:
num |
data_type |
---|---|
1 |
DATA_TYPE_EXT_1N_BYTE |
注意事项:
1、在调用该接口之前必须确保输入的 image 内存已经申请。
2、input output 的 data_type必须相同。
3、目前支持图像的最大width为2048。
代码示例:
int loop =1; int ih = 1080; int iw = 1920; unsigned int ksize = 3; bm_image_format_ext fmt = FORMAT_GRAY; fmt = argc > 1 ? (bm_image_format_ext)atoi(argv[1]) : fmt; ih = argc > 2 ? atoi(argv[2]) : ih; iw = argc > 3 ? atoi(argv[3]) : iw; loop = argc > 4 ? atoi(argv[4]) : loop; ksize = argc >5 ? atoi(argv[5]) : ksize; bm_status_t ret = BM_SUCCESS; bm_handle_t handle; ret = bm_dev_request(&handle, 0); if (ret != BM_SUCCESS) throw("bm_dev_request failed"); bm_image_data_format_ext data_type = DATA_TYPE_EXT_1N_BYTE; bm_image input; bm_image output; bm_image_create(handle, ih, iw, fmt, data_type, &input); bm_image_alloc_dev_mem(input); bm_image_create(handle,ih, iw, fmt, data_type, &output); bm_image_alloc_dev_mem(output); std::shared_ptr<unsigned char*> ch0_ptr = std::make_shared<unsigned char*>(new unsigned char[ih * iw]); std::shared_ptr<unsigned char*> tpu_res_ptr = std::make_shared<unsigned char *>(new unsigned char[ih * iw]); std::shared_ptr<unsigned char*> cpu_res_ptr = std::make_shared<unsigned char *>(new unsigned char[ih*iw]); for (int i = 0; i < loop; i++) { for (int j = 0; j < ih * iw; j++) { (*ch0_ptr.get())[j] = j % 256; } unsigned char *host_ptr[] = {*ch0_ptr.get()}; bm_image_copy_host_to_device(input, (void **)host_ptr); ret = bmcv_image_laplacian(handle, input, output, ksize); if (ret) { cout << "test laplacian failed" << endl; bm_image_destroy(input); bm_image_destroy(output); bm_dev_free(handle); return ret; } else { host_ptr[0] = *tpu_res_ptr.get(); bm_image_copy_device_to_host(output, (void **)host_ptr); } } bm_image_destroy(input); bm_image_destroy(output); bm_dev_free(handle);