Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import matplotlib.pyplot as plt
- import cv2
- import pdb
- testimg = cv2.imread('C:/Users/EPIERSO/Docs/testsimmages/stitch1.jpg',cv2.IMREAD_COLOR)
- testimghsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
- refimg = cv2.imread('C:/Users/EPIERSO/Docs/testsimmages/stitch4.jpg',cv2.IMREAD_COLOR)
- refimghsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
- hbins,sbins = 30,32
- histref = cv2.calcHist([testimghsv],[0,1],histSize =[hbins,sbins],mask = None,ranges=[0,180,0,256])
- histtest = cv2.calcHist([refimghsv],[0,1],histSize =[hbins,sbins],mask = None,ranges=[0,180,0,256])
- signref = np.zeros((hbins*sbins,3),dtype=np.float32)
- signtest = np.zeros((hbins*sbins,3),dtype=np.float32)
- for h in range(hbins):
- for s in range(sbins):
- bin = histref[h,s]
- signref[h*sbins+s,0] = bin
- signref[h*sbins+s,1] = h
- signref[h*sbins+s,2] = s
- bin = hist2[h,s]
- signtest[h*sbins+s,0] = bin
- signtest[h*sbins+s,1] = h
- signtest[h*sbins+s,2] = s
- emd = cv2.EMD(sign1.astype(np.float32),sign2.astype(np.float32),cv2.DIST_L2)
- print(emd)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement