vismatch.im_models.ufm

Classes

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

class vismatch.im_models.ufm.UFMMatcher(device='cpu', max_num_keypoints=1024, min_confidence=0.2, *args, **kwargs)[source][source]

Bases: BaseMatcher

Parameters:

device (str)

preprocess(img)[source][source]
Return type:

Tensor