bmcv_feature_match
该接口用于将网络得到特征点(int8格式)与数据库中特征点(int8格式)进行比对,输出最佳匹配的top-k。
处理器型号支持:
该接口支持BM1684/BM1684X。
接口形式:
bm_status_t bmcv_feature_match( bm_handle_t handle, bm_device_mem_t input_data_global_addr, bm_device_mem_t db_data_global_addr, bm_device_mem_t output_sorted_similarity_global_addr, bm_device_mem_t output_sorted_index_global_addr, int batch_size, int feature_size, int db_size, int sort_cnt = 1, int rshiftbits = 0);
输入参数说明:
bm_handle_t handle
输入参数。bm_handle 句柄。
bm_device_mem_t input_data_global_addr
输入参数。所要比对的特征点数据存储的地址。该数据按照 batch_size * feature_size 的数据格式进行排列。batch_size,feature _size 具体含义将在下面进行介绍。bm_device_mem_t 为内置表示地址的数据类型,可以使用函数 bm_mem_from_system(addr) 将普通用户使用的指针或地址转为该类型,用户可参考示例代码中的使用方式。
bm_device_mem_t db_data_global_addr
输入参数。数据库的特征点数据存储的地址。该数据按照 feature_size * db_size 的数据格式进行排列。feature_size,db_size 具体含义将在下面进行介绍。bm_device_mem_t 为内置表示地址的数据类型,可以使用函数 bm_mem_from_system(addr) 将普通用户使用的指针或地址转为该类型,用户可参考示例代码中的使用方式。
bm_device_mem_t output_sorted_similarity_global_addr
输出参数。每个batch得到的比对结果的值中最大几个值(降序排列)存储地址,具体取多少个值由sort_cnt决定。该数据按照 batch_size * sort_cnt 的数据格式进行排列。batch_size 具体含义将在下面进行介绍。bm_device_mem_t 为内置表示地址的数据类型,可以使用函数 bm_mem_from_system(addr) 将普通用户使用的指针或地址转为该类型,用户可参考示例代码中的使用方式。
bm_device_mem_t output_sorted_index_global_addr
输出参数。每个batch得到的比对结果的在数据库中的序号的存储地址。如对于 batch 0 ,如果 output_sorted_similarity_global_addr 中 bacth 0 的数据是由输入数据与数据库的第800组特征点进行比对得到的,那么 output_sorted_index_global_addr 所在地址对应 batch 0 的数据为800. output_sorted_similarity_global_addr 中的数据按照 batch_size * sort_cnt 的数据格式进行排列。batch_size 具体含义将在下面进行介绍。bm_device_mem_t 为内置表示地址的数据类型,可以使用函数 bm_mem_from_system(addr) 将普通用户使用的指针或地址转为该类型,用户可参考示例代码中的使用方式。
int batch_size
输入参数。待输入数据的 batch 个数,如输入数据有4组特征点,则该数据的 batch_size 为4。batch_size最大值不应超过8。
int feature_size
输入参数。每组数据的特征点个数。feature_size最大值不应该超过4096。
int db_size
输入参数。数据库中数据特征点的组数。db_size最大值不应该超过500000。
int sort_cnt
输入参数。每个 batch 对比结果中所要排序个数,也就是输出结果个数,如需要最大的3个比对结果,则sort_cnt设置为3。该值默认为1。sort_cnt最大值不应该超过30。
int rshiftbits
输入参数。对结果进行右移处理的位数,右移采用round对小数进行取整处理。该参数默认为0。
返回值说明:
BM_SUCCESS: 成功
其他:失败
注意事项:
1、输入数据和数据库中数据的数据类型为 char。
2、输出的比对结果数据类型为 short,输出的序号类型为 int。
3、数据库中的数据在内存的排布为 feature_size * db_size,因此需要将一组特征点进行转置之后再放入数据库中。
4、sort_cnt的取值范围为 1 ~ 30。
示例代码
int batch_size = 4; int feature_size = 512; int db_size = 1000; int sort_cnt = 1; unsigned char src_data_p[4 * 512]; unsigned char db_data_p[512 * 1000]; short output_val[4]; int output_index[4]; for (int i = 0; i < 4 * 512; i++) { src_data_p[i] = rand() % 1000; } for (int i = 0; i < 512 * 1000; i++) { db_data_p[i] = rand() % 1000; } bmcv_feature_match(handle, bm_mem_from_system(src_data_p), bm_mem_from_system(db_data_p), bm_mem_from_system(output_val), bm_mem_from_system(output_index), batch_size, feature_size, db_size, sort_cnt, 8);