import wx
import os
from PIL import Image
from predict import predict #这是我的py文件#主框架
class my_frame(wx.Frame):def __init__(self, parent, title):wx.Frame.__init__(self, parent, title=title, size=(500, 300))self.Show(True)#按钮区域panel1 = wx.Panel(self,pos = (170,330), size=(100, 40))#文本区域panel2 = wx.Panel(self,pos = (580,0), size=(660, 630))#按钮区域panel4 = wx.Panel(self,pos = (350,330), size=(100, 40))#两个按钮b1 = wx.Button(panel1, label="选择识别图片", size=(100, 40))b2 = wx.Button(panel4, label="开始识别图片", size=(100, 40))#按钮关联操作b1.Bind(wx.EVT_BUTTON, self.on_open)b2.Bind(wx.EVT_BUTTON, self.detect)#文本区域相关操作vbox2 = wx.BoxSizer(wx.VERTICAL)self.text = wx.TextCtrl(panel2, style=wx.TE_MULTILINE, size=(660, 630))font = wx.Font(15, wx.MODERN, wx.NORMAL, wx.NORMAL, False, 'Consolas')self.text.SetFont(font)vbox2.Add(self.text, 1, flag=wx.ALL | wx.EXPAND, border=10)panel2.SetSizer(vbox2)#返回识别结果def detect(self, e):result,time = predict(path)self.text.AppendText(u"===================================\n")self.text.AppendText(u"用时{:.5}s,".format(time) + "识别结果:" +'\n'+result+'\n')self.text.AppendText(u"===================================\n")#打开文件def on_open(self, e):dlg = wx.FileDialog(self, u"请选择一个图片", 'mnist_test', "", "*.*", wx.FD_OPEN)# 调用一个函数打开对话框if dlg.ShowModal() == wx.ID_OK:#这里有个问题,就是图片显示在panel区域后出现打开新一张图片#但上一张还存在的现象,因为没找到解决方法没所以就,把图片调成一样大小,盖住....#可以不把图片显示在一个frame里,这样就可以随时关闭打开的图片image = wx.Image(dlg.GetPath()).Rescale(400, 50)#显示图片在一个新paneltemp = image.ConvertToBitmap()Size = temp.GetWidth(), temp.GetHeight()panel3 = wx.Panel(self,pos = (120,110), size=Size)bmp = wx.StaticBitmap(parent=panel3, bitmap=temp)#这里设置全局变量是因为上面的函数要用这个变量global pathpath = dlg.GetPath()dlg.Destroy()app = wx.App()
frame = my_frame(None, '基于TensorFlow的CRNN文本识别')
app.MainLoop()
效果:
总结
颇有一种东拼西凑质感。。。
算了,继续加油。