vismatch.im_models.aspanformer

Classes

AspanformerMatcher([device])

class vismatch.im_models.aspanformer.AspanformerMatcher(device='cpu', **kwargs)[source][source]

Bases: BaseMatcher

Parameters:: device (str)

hf_model_id = 'vismatch/aspanformer'

weights_src = 'https://drive.google.com/file/d/1eavM9dTkw9nbc-JqlVVfGPU5UvTTfc6k/view'

divisible_size = 32

download_weights(cache_dir)[source][source]

preprocess(img)[source][source]