gbarbarani / difftda_pytorch Goto Github PK

Hello, @ZerothB @gbarbarani ,-I really appreciate for your "re-ranking-for-VPR" work！Congratulations :)
I need your help with these questions：

一、When I run re-ranking, I encounter the following problems:
1、D2net
Error info:
Traceback (most recent call last):
File "scripts/rerank.py", line 76, in
rerankend_predictions, reranked_scores = extract_reranked_prediction(extractor,
File "/home/1/Desktop/re-ranking-for-VPR-main_8.3/src/utils/utils.py", line 186, in extract_reranked_prediction
query_out = extractor({"image": query_image})
File "/home/1/anaconda3/envs/VPR2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/home/1/Desktop/re-ranking-for-VPR-main_8.3/src/models/d2net.py", line 59, in forward
keypoints, scores, descriptors = process_multiscale(
File "/home/1/Desktop/re-ranking-for-VPR-main_8.3/src/models/../../third_party/d2-net/lib/pyramid.py", line 86, in process_multiscale
fmap_pos = fmap_pos[:, ids]
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

2、Patchnetvlad_MultiScaleRANSAC
Error info:
Traceback (most recent call last):
File "scripts/rerank.py", line 60, in
scorer = eval(args.scorer).to(args.device).eval()
File "", line 1, in
TypeError: init() takes 1 positional argument but 2 were given

3、Patchnetvlad_ Rapid
Error info:
Traceback (most recent call last):
File "scripts/rerank.py", line 60, in
scorer = eval(args.scorer).to(args.device).eval()
File "", line 1, in
NameError: name 'Rapid' is not defined

How can I solve these three problems?

二、I noticed that there are also DELG+RANSAC, DELG+RRT methods in the paper, but these parts are not found in the benchmark.sh,is it convenient to provide relevant codes?Or are these two parts(DELG+RANSAC, DELG+RRT) difficult to integrate into the existing code repository?

I am always looking forward to your kind response.
Best regards.