vismatch.im_models.topicfm

Classes

TopicFMMatcher([device, variant])

class vismatch.im_models.topicfm.TopicFMMatcher(device='cpu', variant='fast', *args, **kwargs)

Bases: BaseMatcher

Parameters:

device (str)

divisible_size = 16

preprocess(img)

Convert RGB image to grayscale, resize to divisible, and add batch dimension.