使用示例
本章节以使用sift-128-euclidean.hdf5数据集为例,提供通过Python调用KScaNN算法接口的完整使用示例。
获取测试代码
请参见3获取的源代码中的测试框架代码,假设源代码位于“/path/to/scann/sra_scann_adapter”,测试框架代码位于“/path/to/scann/sra_scann_adapter/ann-benchmarks”下。获取数据集。
1 2 3 | cd /path/to/scann/sra_scann_adapter/ann-benchmarks mkdir data && cd data wget http://ann-benchmarks.com/sift-128-euclidean.hdf5 --no-check-certificate |
主要文件的目录结构如下所示:
1 2 3 4 5 6 7 | ├── data // 存放数据集 └── sift-128-euclidean.hdf5 ├── ann_benchmarks └── algorithms └── scann └── config-sift-128-euclidean.yml // 对应数据集配置文件 └── test.sh // 测试脚本 |
使用步骤
- 请确保12已安装scann-1.2.10-cp39-cp39-linux_aarch64.whl。
- 安装ann-benchmarks所需依赖。
1 2 3
cd /path/to/scann/sra_scann_adapter/ann-benchmarks pip install -r requirements.txt yum install numactl numactl-devel
- 运行测试脚本。
1sh test.sh
运行结果如下:

父主题: Python接口