test_matcher.py 346 B

123456789101112
  1. import numpy as np
  2. from superpoint_superglue_deployment import Matcher
  3. def test_inference():
  4. query_image = np.random.rand(300, 300) * 255
  5. query_image = query_image.astype(np.uint8)
  6. ref_image = query_image.astype(np.uint8)
  7. ref_image = ref_image.astype(np.uint8)
  8. matcher = Matcher()
  9. matcher.match(query_image, ref_image)