OpenCV 实现滑块缺口位置计算

maiyantao

1. import io
   
2. import cv2
   
3. import numpy as np
   
4. from PIL import Image
   
5. 
   
6. 
   
7. class DetectDistanceXY:
   
8. 
   
9.     def detectDistanceX(self, imgSlider, imgBackground):
   
10.         """X坐标缺口检测"""
    
11. 
    
12.         # imgSlider = cv2.cvtColor(np.array(Image.open(io.BytesIO(imgSlider))), cv2.COLOR_BGR2GRAY)
    
13.         # imgBackground = cv2.cvtColor(np.array(Image.open(io.BytesIO(imgBackground))), cv2.COLOR_BGR2GRAY)
    
14. 
    
15.         imgSliderGray = cv2.cvtColor(np.array(Image.open(imgSlider)), cv2.COLOR_BGR2GRAY)
    
16.         imgBackgroundGray = cv2.cvtColor(np.array(Image.open(imgBackground)), cv2.COLOR_BGR2GRAY)
    
17.         # 寻找最佳匹配
    
18.         res = cv2.matchTemplate(self._tran_canny(imgSliderGray), self._tran_canny(imgBackgroundGray), cv2.TM_CCOEFF_NORMED)
    
19.         # 最小值，最大值，并得到最小值, 最大值的索引
    
20.         min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)
    
21. 
    
22.         return max_loc[0]
    
23. 
    
24.     def detectDistanceY(self, imgSlider, imgBackground):
    
25.         #  传入二进制图片
    
26.         # target_gray = cv2.cvtColor(np.array(Image.open(io.BytesIO(imgSlider))), cv2.COLOR_BGR2GRAY)
    
27.         # template_gray = cv2.cvtColor(np.array(Image.open(io.BytesIO(imgBackground))), cv2.COLOR_BGR2GRAY)
    
28.         # 根据路径读取图片
    
29.         imgSliderGray = cv2.cvtColor(np.array(Image.open(imgSlider)), cv2.COLOR_BGR2GRAY)
    
30.         imgBackgroundGray = cv2.cvtColor(np.array(Image.open(imgBackground)), cv2.COLOR_BGR2GRAY)
    
31.         res = cv2.matchTemplate(imgSliderGray, imgBackgroundGray, cv2.TM_CCOEFF_NORMED)
    
32.         min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)
    
33.         return max_loc[0]
    
34. 
    
35.     def _tran_canny(self, image):
    
36.         """消除噪声"""
    
37. 
    
38.         image = cv2.GaussianBlur(image, (3, 3), 0)
    
39.         return cv2.Canny(image, 100, 200)
    
40. 
    
41. 
    
42. if __name__ == '__main__':
    
43.     detectdistancexy = DetectDistanceXY()
    
44. 
    
45.     targetFile = r'target.png'
    
46.     templateFile = r'template.png'
    
47.     print(detectdistancexy.detectDistanceX(targetFile, templateFile))
    
48.     print(detectdistancexy.detectDistanceY(targetFile, templateFile))

复制代码