
基于PyTorch的DELG:整合深度局部与全局特征的图像检索方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了基于PyTorch框架实现的DELG模型,该模型创新性地结合了深度学习技术中的局部和全局特征,显著提升了图像检索的效果。
DELG-Pytorch 是一个用于统一深层本地和全局特征以进行图像搜索的Pytorch实现。以下是 DELG pipline 的安装步骤:
1. 安装Python依赖项:
```
pip install -r requirements.txt
```
2. 设置PYTHONPATH环境变量:
```
export PYTHONPATH=`pwd`:$PYTHONPATH
```
3. 训练挖掘模型:
```
python train_delg.py \
--cfg configs/metric/resnet_delg_8gpu.yaml \
OUT_DIR ./output \
PORT 12001 \
TRAIN.WEIGHTS path/to/pretrainedmodel
```
4. 提取多尺度的全局和局部特征:
```
python tools/extractor.py --cfg configs/resnet_de
```
全部评论 (0)
还没有任何评论哟~


