vismatch.im_models.se2loftr

Classes

Se2LoFTRMatcher([device, max_num_keypoints, ...])

class vismatch.im_models.se2loftr.Se2LoFTRMatcher(device='cpu', max_num_keypoints=0, loftr_config='rot8', *args, **kwargs)

Bases: BaseMatcher

Parameters:

device (str)

configs = {'big': src.config.default._CN.clone, 'dense': src.config.default._CN.clone, 'rot4': src.config.default._CN.clone, 'rot8': src.config.default._CN.clone}

weights_filenames = {'big': 'se2loftr_rot4_big.safetensors', 'dense': 'se2loftr_rot4_dense.safetensors', 'rot4': 'se2loftr_rot4.safetensors', 'rot8': 'se2loftr_rot8.safetensors'}

divisible_size = 32

load_model(config, device='cpu')

preprocess(img)