from PIL import Image
import cv2
import numpy as np
#import skimage.io as io
import matplotlib.pyplot as plt
#filename="/home/jiazhe"
#str=filename+"/*.jpg"
#mat=io.ImageCollection(str)
def drawMatch_cv2(im1,kp1,im2,kp2,goodMatch):
h1,w1=im1.shape[:2]
h2,w2=im2.shape[:2]
aa=np.zeros((max(h1,h2),w1+w2,3),np.uint8)
aa[:h1,:w1]=im1
aa[:h2,w1:w1+w2]=im2
p1=[kpp.queryIdx for kpp in goodMatch]
p2=[kpp.trainIdx for kpp in goodmatch]
post1=np.int32([kp1[pp].pt for pp in p1])
post2=np.int32([kp2[pp].pt for pp in p2])+(w1,0)
for(x1,y1),(x2,y2) in zip(post1,post2):
cv2.line(aa,(x1,y1),(x2,y2),(0,0,255))
cv2.namedWindow("match",cv2.WINDOW_NORMAL)
cv2.imshow("match",vis)
im1=cv2.imread("a.jpg")
im2=cv2.imread("b.jpg")
#img=cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
vis1 = im1.copy()
vis2 = im2.copy()
#img=mat(0)
mser=cv2.MSER_create(_min_area=1200)
regions1,boxes1=mser.detectRegions(vis1)
regions2,boxes2=mser.detectRegions(vis2)
kpkp1=mser.detect(vis1)
kpkp2=mser.detect(vis2)
#print len(mser.detect(vis))
for i in range(len(boxes1)):
x,y,w,h=boxes1[i]
cv2.rectangle(vis1,(x,y),(x+w,y+h),(255,0,0),2)
siftt=cv2.xfeatures2d.SIFT_create()
kp1=siftt.detect(vis1,None)
des1=siftt.compute(vis1,kpkp1)
for i in range(len(boxes2)):
x,y,w,h=boxes2[i]
cv2.rectangle(vis2,(x,y),(x+w,y+h),(255,0,0),2)
siftt=cv2.xfeatures2d.SIFT_create()
#print len(regions)
#print len(boxes)
kp2=siftt.detect(vis2,None)
des2=siftt.compute(vis2,kpkp2)
#print len(des[0])
cv2.drawKeypoints(vis1,kpkp1,im1,(0,255,0))
cv2.drawKeypoints(vis2,kpkp2,im2,(0,255,0))
bf=cv2.BFMatcher(cv2.NORM_L2)
matches=bf.knnMatch(des1,des2,k=2)
goodMatch=[]
for m,n in matches:
if m.distance<0.5*n.distance:
goodMatch.append(m)
drawMatch_cv2(im1,kp1,im2,kp2,goodMatch[:20])
cv2.waitKey(0)
cv2.destroyAllWindows()