Spaces:
Running
Running
File size: 1,503 Bytes
9cde3b4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
from PIL import Image
import numpy as np
import numpy as np
import cv2 as cv
import matplotlib.pyplot as plt
if __name__ == "__main__":
from argparse import ArgumentParser
parser = ArgumentParser()
parser.add_argument("--im_A_path", default="assets/toronto_A.jpg", type=str)
parser.add_argument("--im_B_path", default="assets/toronto_B.jpg", type=str)
parser.add_argument("--save_path", default="demo/roma_warp_toronto.jpg", type=str)
args, _ = parser.parse_known_args()
im1_path = args.im_A_path
im2_path = args.im_B_path
save_path = args.save_path
img1 = cv.imread(im1_path,cv.IMREAD_GRAYSCALE) # queryImage
img2 = cv.imread(im2_path,cv.IMREAD_GRAYSCALE) # trainImage
# Initiate SIFT detector
sift = cv.SIFT_create()
# find the keypoints and descriptors with SIFT
kp1, des1 = sift.detectAndCompute(img1,None)
kp2, des2 = sift.detectAndCompute(img2,None)
# BFMatcher with default params
bf = cv.BFMatcher()
matches = bf.knnMatch(des1,des2,k=2)
# Apply ratio test
good = []
for m,n in matches:
if m.distance < 0.75*n.distance:
good.append([m])
# cv.drawMatchesKnn expects list of lists as matches.
draw_params = dict(matchColor = (255,0,0), # draw matches in red color
singlePointColor = None,
flags = 2)
img3 = cv.drawMatchesKnn(img1,kp1,img2,kp2,good,None,**draw_params)
Image.fromarray(img3).save("demo/sift_matches.png")
|