(slide_img, bg_img, check=False)
| 33 | |
| 34 | |
| 35 | def find_location(slide_img, bg_img, check=False): |
| 36 | # 读取背景图和滑块图 |
| 37 | bg = cv2.imread(bg_img) |
| 38 | slice = cv2.imread(slide_img) |
| 39 | |
| 40 | # 对图片做灰度处理 |
| 41 | bg = cv2.cvtColor(bg, cv2.COLOR_BGR2GRAY) |
| 42 | slice = cv2.cvtColor(slice, cv2.COLOR_BGR2GRAY) |
| 43 | |
| 44 | # 对图片进行边缘处理 |
| 45 | bg = cv2.Canny(bg, 100, 200) |
| 46 | slice = cv2.Canny(slice, 100, 200) |
| 47 | |
| 48 | # 匹配图片相似度 |
| 49 | result = cv2.matchTemplate(bg, slice, cv2.TM_CCOEFF_NORMED) |
| 50 | min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(result) |
| 51 | |
| 52 | if check: |
| 53 | cv2.rectangle(bg, (max_loc[0], max_loc[1]), (max_loc[0] + 50, max_loc[1] + 50), (255, 255, 255)) |
| 54 | cv2.imshow('bg', bg) |
| 55 | cv2.imshow('slice', slice) |
| 56 | cv2.waitKey(0) |
| 57 | cv2.destroyAllWindows() |
| 58 | |
| 59 | return max_loc[0] |
| 60 | |
| 61 | |
| 62 | def base64_to_image(base64_str, image_path): |
no outgoing calls
no test coverage detected