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)
#显示图片在一个新panel
temp = image.ConvertToBitmap()
Size = temp.GetWidth(), temp.GetHeight()
panel3 = wx.Panel(self,pos = (120,110), size=Size)
bmp = wx.StaticBitmap(parent=panel3, bitmap=temp)
#这里设置全局变量是因为上面的函数要用这个变量
global path
path = dlg.GetPath()
dlg.Destroy()
app = wx.App()
frame = my_frame(None, '基于TensorFlow的CRNN文本识别')
app.MainLoop()
效果:
总结
颇有一种东拼西凑质感。。。
算了,继续加油。