深度局部特征(DeLF)的Pytorch实现 PyTorch实现的“具有深深的局部特征的大规模图像检索” 参考: : 先决条件 火炬 python3 CUDA 训练DeLF DeLF训练有两个步骤:(1)调整阶段,和(2)关键点阶段。 Finetune阶段加载ImageNet上预训练的resnet50模型,并进行优化。 关键点阶段将冻结“基本”网络,并且仅更新“注意”网络以进行关键点选择。 训练过程完成后,模型将保存在repo//keypoint/ckpt (1)培训微调阶段: $ cd train/ $ python main.py \ --stage ' finetune ' \ --optim ' sgd ' \ --gpu_id 6 \ --expr ' landmark ' \ --ncls 586 \ --f
1