Oxford5K, Paris6K, ROxford5K, RParis6K
收藏资源简介:
该仓库包含多个用于图像检索的数据集,包括Oxford5K, Paris6K, ROxford5K和RParis6K。这些数据集用于训练和评估深度神经网络模型,支持自动下载和转换为kapture格式。
This repository contains multiple datasets for image retrieval, including Oxford5K, Paris6K, ROxford5K and RParis6K. These datasets are used for training and evaluating deep neural network models, and support automatic download and conversion to the kapture format.
Neural Image Retrival 数据集概述
评估
-
评估脚本:
test_dir.py用于评估预训练模型并重现结果。 -
参数:
--dataset: 选择数据集(例如:Oxford5K, Paris6K, ROxford5K, RParis6K)[必需]--checkpoint: 模型权重路径 [必需]--whiten: 对输出特征应用白化 [默认 Landmarks_clean]--whitenp: 白化功率 [默认: 0.25]--aqe: alpha-query 扩展参数 [默认: None]--trfs: 输入图像变换(可用于应用多尺度)[默认: None]--gpu: 选择 GPU ID(-1 选择 CPU)
-
示例命令: bash python -m dirtorch.test_dir --dataset RParis6K --checkpoint dirtorch/data/Resnet101-AP-GeM.pt --whiten Landmarks_clean --whitenp 0.25 --gpu 0
-
输出示例:
Evaluation...
- mAP-easy = 0.907568
- mAP-medium = 0.803098
- mAP-hard = 0.608556
特征提取与 kapture 数据集
-
安装 kapture: bash pip install kapture
-
下载数据集: bash kapture_download_dataset.py update kapture_download_dataset.py list
例如:安装 Extended-CMU-Seasons_slice22 的 mapping 和 query
kapture_download_dataset.py install "Extended-CMU-Seasons_slice22_*"
-
提取 kapture 数据集的全局特征: bash python -m dirtorch.extract_kapture --kapture-root pathto/yourkapturedataset --checkpoint dirtorch/data/Resnet101-AP-GeM-LM18.pt --gpu 0
-
更多信息: 运行
python -m dirtorch.extract_kapture --help获取提取参数的详细信息。




