Spaces:
Runtime error
Runtime error
File size: 1,054 Bytes
1bb1365 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# Copyright (C) 2024-present Naver Corporation. All rights reserved.
# Licensed under CC BY-NC-SA 4.0 (non-commercial use only).
#
# --------------------------------------------------------
# Cambridge Landmarks dataloader
# --------------------------------------------------------
import os
from dust3r_visloc.datasets.base_colmap import BaseVislocColmapDataset
class VislocCambridgeLandmarks(BaseVislocColmapDataset):
def __init__(self, root, subscene, pairsfile, topk=1, cache_sfm=False):
image_path = os.path.join(root, subscene)
map_path = os.path.join(root, "mapping", subscene, "colmap/reconstruction")
query_path = os.path.join(root, "kapture", subscene, "query")
pairsfile_path = os.path.join(
root, subscene, "pairsfile/query", pairsfile + ".txt"
)
super().__init__(
image_path=image_path,
map_path=map_path,
query_path=query_path,
pairsfile_path=pairsfile_path,
topk=topk,
cache_sfm=cache_sfm,
)
|