帮我优化我下面发送的代码,使图片识别过程中在屏幕上绘制出图片位置import tkinter as tk from tkinter import filedialog import pyautogui class Application(tk.Frame): def __init__(self, master=None): super().__init__(master) self.master = master self.pack() self.create_widgets() def create_widgets(self): self.select_button = tk.Button(self) self.select_button["text"] = "选择图片" self.select_button["command"] = self.select_image self.select_button.pack(side="top") self.result_label = tk.Label(self) self.result_label.pack(side="bottom") def select_image(self): # 弹出文件选择对话框选择要查找的图片 file_path = filedialog.askopenfilename(title="选择图片文件", filetypes=[("Image files", "*.png;*.jpg;*.jpeg")]) # 如果选择了图片则查找图片位置 if file_path: location = find_image_location(file_path) if location is not None: self.result_label["text"] = f"图片 {file_path} 的位置坐标为:{location}" else: self.result_label["text"] = f"图片 {file_path} 不存在于桌面" def find_image_location(image_path): # 加载图片并获取图片位置坐标 location = pyautogui.locateOnScreen(image_path,confidence=0.9) if location is not None: # 获取图片中心点的坐标 x, y = pyautogui.center(location) return (x, y) else: return None root = tk.Tk() app = Application(master=root) app.mainloop()
时间: 2024-02-14 20:07:36 浏览: 37
好的,你可以使用 pyautogui 库的 `screenshot()` 和 `drawRect()` 函数来实现在屏幕上绘制图片位置。
具体实现过程如下:
1. 在 find_image_location 函数中,在使用 pyautogui 库的 locateOnScreen 函数查找图片位置之前,先使用 pyautogui 库的 screenshot 函数获取当前屏幕的截图,并将其保存到变量 im 中。
2. 在使用 pyautogui 库的 locateOnScreen 函数查找图片位置之后,如果找到了图片,则使用 pyautogui 库的 drawRect 函数在屏幕上绘制出图片的位置矩形框。
3. 在使用 pyautogui 库的 drawRect 函数绘制矩形框时,需要传入图片位置的左上角坐标和右下角坐标,可以使用 pyautogui 库的 left、top、width 和 height 属性计算得到。
修改后的代码如下:
```
import tkinter as tk
from tkinter import filedialog
import pyautogui
class Application(tk.Frame):
def __init__(self, master=None):
super().__init__(master)
self.master = master
self.pack()
self.create_widgets()
def create_widgets(self):
self.select_button = tk.Button(self)
self.select_button["text"] = "选择图片"
self.select_button["command"] = self.select_image
self.select_button.pack(side="top")
self.result_label = tk.Label(self)
self.result_label.pack(side="bottom")
def select_image(self):
# 弹出文件选择对话框选择要查找的图片
file_path = filedialog.askopenfilename(title="选择图片文件", filetypes=[("Image files", "*.png;*.jpg;*.jpeg")])
# 如果选择了图片则查找图片位置
if file_path:
location = find_image_location(file_path)
if location is not None:
self.result_label["text"] = f"图片 {file_path} 的位置坐标为:{location}"
else:
self.result_label["text"] = f"图片 {file_path} 不存在于桌面"
def find_image_location(image_path):
# 获取当前屏幕截图
im = pyautogui.screenshot()
# 加载图片并获取图片位置坐标
location = pyautogui.locateOnScreen(image_path, confidence=0.9)
if location is not None:
# 获取图片位置的左上角和右下角坐标
left, top, width, height = location
right = left + width
bottom = top + height
# 在屏幕上绘制图片位置矩形框
pyautogui.drawRect(left, top, width, height)
# 获取图片中心点的坐标
x, y = pyautogui.center(location)
return (x, y)
else:
return None
root = tk.Tk()
app = Application(master=root)
app.mainloop()
```
这样,当程序找到图片位置时,会在屏幕上绘制出图片位置的矩形框,方便用户直观地了解图片位置。