Skip to content

Commit a3e5731

Browse files
authored
Adapt to pycolmap.Sift interface change from the new pycolmap update. (#484)
1 parent 624866f commit a3e5731

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

hloc/extractors/dog.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,9 @@ def _forward(self, data):
5757
else:
5858
options["normalization"] = pycolmap.Normalization.L2
5959
self.sift = pycolmap.Sift(
60-
options=pycolmap.SiftExtractionOptions(options),
60+
options=pycolmap.FeatureExtractionOptions(
61+
sift=pycolmap.SiftExtractionOptions(options)
62+
),
6163
device=getattr(pycolmap.Device, "cuda" if use_gpu else "cpu"),
6264
)
6365

0 commit comments

Comments
 (0)