vismatch.im_models.matching_toolbox

Classes

D2netMatcher([device])

DogAffHardNNMatcher([device, max_num_keypoints])

Patch2pixMatcher([device])

R2D2Matcher([device, max_num_keypoints])

SuperGlueMatcher([device, max_num_keypoints])

class vismatch.im_models.matching_toolbox.Patch2pixMatcher(device='cpu', *args, **kwargs)[source][source]

Bases: BaseMatcher

Parameters:

device (str)

divisible_by = 32
preprocess(img)[source][source]
class vismatch.im_models.matching_toolbox.SuperGlueMatcher(device='cpu', max_num_keypoints=2048, *args, **kwargs)[source][source]

Bases: BaseMatcher

Parameters:

device (str)

SG_WEIGHTS_URL = 'https://github.com/magicleap/SuperGluePretrainedNetwork/blob/c0626d58c843ee0464b0fa1dd4de4059bfae0ab4/models/weights/superglue_MODEL_VERSION.pth?raw=true'
SP_WEIGHTS_URL = 'https://github.com/magicleap/SuperGluePretrainedNetwork/blob/c0626d58c843ee0464b0fa1dd4de4059bfae0ab4/models/weights/superpoint_v1.pth?raw=true'
SG_WEIGHTS_DIR = PosixPath('/home/docs/checkouts/readthedocs.org/user_builds/vismatch/checkouts/stable/vismatch/third_party/imatch-toolbox/third_party/superglue/models/weights')
class vismatch.im_models.matching_toolbox.R2D2Matcher(device='cpu', max_num_keypoints=2048, *args, **kwargs)[source][source]

Bases: BaseMatcher

Parameters:

device (str)

class vismatch.im_models.matching_toolbox.D2netMatcher(device='cpu', *args, **kwargs)[source][source]

Bases: BaseMatcher

Parameters:

device (str)

static preprocess(img_tensor)[source][source]
class vismatch.im_models.matching_toolbox.DogAffHardNNMatcher(device='cpu', max_num_keypoints=2048, *args, **kwargs)[source][source]

Bases: BaseMatcher

Parameters:

device (str)

static tensor_to_numpy_int(im_tensor)[source][source]