File size: 698 Bytes
3ef85e9 |
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 2022-present NAVER Corp.
# CC BY-NC-SA 4.0
# Available only for non-commercial use
from pdb import set_trace as bb
from os.path import *
from .image_set import ImageSet
from .pair_dataset import UnsupervisedPairs
class SfM120k_Images (ImageSet):
def __init__(self, root='datasets/sfm120k'):
self.init_from_folder(join(root,'ims'), recursive=True, listing=True, exts='')
class SfM120k_Pairs (UnsupervisedPairs):
def __init__(self, root='datasets/sfm120k'):
super().__init__(SfM120k_Images(root=root), join(root,'list_pairs.txt'))
if __name__ == '__main__':
from tools.viz import show_random_pairs
db = SfM120k_Pairs()
show_random_pairs(db)
|