from PIL import Image import tkinter as tk def show_results(results): # 创建子界面 win = tk.Toplevel() win.geometry("400x400") win.title("Results") # 创建表格 table = tk.Frame(win) table.pack() # 创建表头 tk.Label(table, text="X").grid(row=0, column=0) tk.Label(table, text="Y").grid(row=0, column=1) tk.Label(table, text="W").grid(row=0, column=2) tk.Label(table, text="H").grid(row=0, column=3) tk.Label(table, text="类别").grid(row=0, column=4) # 创建表格内容 for i, s1 in enumerate(results): tk.Label(table, text=s1[0]).grid(row=i + 1, column=0) tk.Label(table, text=s1[1]).grid(row=i + 1, column=1) tk.Label(table, text=s1[2]).grid(row=i + 1, column=2) tk.Label(table, text=s1[3]).grid(row=i + 1, column=3) tk.Label(table, text=s1[4]).grid(row=i + 1, column=4) # 创建选择按钮 select_button = tk.Button(table, text="选择", command=lambda s=s1: select_result(s)) select_button.grid(row=i + 1, column=5) # 定义选择结果函数 def select_result(result): print("选择的是:", result) def site(source, pred, names): img = Image.open(source) x1, x2 = img.size print([x1, x2]) results = [] for i1 in pred: s = [] for i2 in i1.data.cpu().numpy(): s1 = [] s = list(i2) # 获取预测框中心点的坐标 x = s[0] = float(round((s[0] + s[2]) / 2 / x1, 4)) y = s[1] = float(round((s[1] + s[3]) / 2 / x2, 4)) # 预测框的宽和高 w = s[2] - s[0] h = s[3] - s[1] s1.append(str(x)) s1.append(str(y)) s1.append(str(w)) s1.append(str(h)) s1.append(names[int(s[5])]) if s[4] < 0.5: break results.append(s1) # 创建GUI界面 window = tk.Tk() window.geometry("400x400") # 创建按钮 for name in set([r[4] for r in results]): tk.Label(window, text=name).pack() button = tk.Button(window, text="显示" + name + " 的结果", command=lambda name=name: show_results([r[:4]+[name] for r in results if r[4] == name])) button.pack() # 创建确定按钮 confirm_button = tk.Button(window, text="退出", command=window.quit) confirm_button.pack() window.mainloop() 通过这个程序计算出来的预测框的xy宽高对不对
时间: 2024-02-03 10:02:01 浏览: 39
我无法确定预测框的xy宽高是否正确,因为我无法看到输入数据和模型的输出结果。建议你在程序中添加一些输出语句,以便检查预测结果是否正确。例如,在site函数中,你可以添加以下代码来输出每个预测框的坐标信息:
```
print("预测框坐标信息:", s1)
```
然后在show_results函数中,你可以添加以下代码来检查传递给该函数的results列表是否正确:
```
print("传递给show_results函数的results列表:", results)
```
这样可以帮助你确定预测框的xy宽高是否正确。
阅读全文