We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 624866f commit a3e5731Copy full SHA for a3e5731
hloc/extractors/dog.py
@@ -57,7 +57,9 @@ def _forward(self, data):
57
else:
58
options["normalization"] = pycolmap.Normalization.L2
59
self.sift = pycolmap.Sift(
60
- options=pycolmap.SiftExtractionOptions(options),
+ options=pycolmap.FeatureExtractionOptions(
61
+ sift=pycolmap.SiftExtractionOptions(options)
62
+ ),
63
device=getattr(pycolmap.Device, "cuda" if use_gpu else "cpu"),
64
)
65
0 commit comments