BRISK keypoint detection

brisk.py

import cv2 as cv

type Image = cv.typing.MatLike

def show_image(img: Image):
  cv.imshow("out", img)
  cv.waitKey(0)
  cv.destroyAllWindows()

def brisk_compare(train_img: Image, query_img: Image):
  Brisk = cv.BRISK_create()

  kp_train, desc_train = Brisk.detectAndCompute(train_img, None)
  kp_query, desc_query = Brisk.detectAndCompute(query_img, None)
  
  # brute force matcher
  BFMatcher = cv.BFMatcher(normType = cv.NORM_HAMMING, crossCheck = True)
  matches = BFMatcher.match(queryDescriptors = desc_train, trainDescriptors = desc_query)

  # only draw the top 50 matches
  top_n = 50
  matches = sorted(matches, key = lambda x: x.distance)
  output = cv.drawMatches(img1 = train_img, keypoints1 = kp_train,
                          img2 = query_img, keypoints2 = kp_query,
                          matches1to2 = matches[:top_n],
                          matchColor = (0,255,0),
                          outImg = None)
  return output

def main():
  train_img = cv.imread('fosforos_train.jpg')
  query_img = cv.imread('fosforos_query.jpg')
  out_img = brisk_compare(train_img, query_img)
  cv.imwrite("brisk_out.jpg", out_img)
  show_image(out_img)

main()

brisk_out.jpg