# -*- coding: utf-8 -*- import wx import modelmngr_execute ########################################################################################## class MyFrame1 ( wx.Frame ): def __init__( self, parent ): wx.Frame.__init__ ( self, parent, id = wx.ID_ANY, title = u"模型管家", pos = wx.DefaultPosition, size = wx.Size( 779,521 ), style = wx.DEFAULT_FRAME_STYLE|wx.TAB_TRAVERSAL ) self.SetSizeHintsSz( wx.DefaultSize, wx.DefaultSize ) #################################################### bSizer6 = wx.BoxSizer( wx.VERTICAL ) #################################################### bSizer7 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText6 = wx.StaticText( self, wx.ID_ANY, u"模型文件1路径 *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText6.Wrap( -1 ) bSizer7.Add( self.m_staticText6, 0, wx.ALL, 5 ) self.m_textCtrl6 = wx.TextCtrl( self, wx.ID_ANY, u"/system/etc/models/fsrcn.cambricon", wx.DefaultPosition, wx.Size( 650,-1 ), 0 ) bSizer7.Add( self.m_textCtrl6, 0, wx.ALL, 5 ) bSizer6.Add( bSizer7, 1, wx.EXPAND, 5 ) ######################## bSizer90 = wx.BoxSizer( wx.HORIZONTAL ) self.m_checkBox2 = wx.CheckBox( self, wx.ID_ANY, u"是否加密 *", wx.DefaultPosition, wx.DefaultSize, 0 ) bSizer90.Add( self.m_checkBox2, 0, wx.ALL, 5 ) self.Bind(wx.EVT_CHECKBOX, self.on_model1_check, self.m_checkBox2) self.m_staticText19 = wx.StaticText( self, wx.ID_ANY, u"模型文件1密钥", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText19.Wrap( -1 ) bSizer90.Add( self.m_staticText19, 0, wx.ALL, 5 ) self.m_textCtrl19 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 300,-1 ), 0 ) bSizer90.Add( self.m_textCtrl19, 0, wx.ALL, 5 ) bSizer6.Add( bSizer90, 1, wx.EXPAND, 5 ) #################################################### bSizer24 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText20 = wx.StaticText( self, wx.ID_ANY, u"模型文件2路径", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText20.Wrap( -1 ) bSizer24.Add( self.m_staticText20, 0, wx.ALL, 5 ) self.m_textCtrl20 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 650,-1 ), 0 ) bSizer24.Add( self.m_textCtrl20, 0, wx.ALL, 5 ) bSizer6.Add( bSizer24, 1, wx.EXPAND, 5 ) ######################## bSizer12 = wx.BoxSizer( wx.HORIZONTAL ) self.m_checkBox3 = wx.CheckBox( self, wx.ID_ANY, u"是否加密", wx.DefaultPosition, wx.DefaultSize, 0 ) bSizer12.Add( self.m_checkBox3, 0, wx.ALL, 5 ) self.Bind(wx.EVT_CHECKBOX, self.on_model2_check, self.m_checkBox3) self.m_staticText21 = wx.StaticText( self, wx.ID_ANY, u"模型文件2密钥", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText21.Wrap( -1 ) bSizer12.Add( self.m_staticText21, 0, wx.ALL, 5 ) self.m_textCtrl21 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 300,-1 ), 0 ) bSizer12.Add( self.m_textCtrl21, 0, wx.ALL, 5 ) bSizer6.Add( bSizer12, 1, wx.EXPAND, 5 ) #################################################### bSizer31 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText22 = wx.StaticText( self, wx.ID_ANY, u"模型类型 *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText22.Wrap( -1 ) bSizer31.Add( self.m_staticText22, 0, wx.ALL, 5 ) m_choice1Choices = [ u"在线模型(0)", u"离线模型(1)" ] self.m_choice1 = wx.Choice( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, m_choice1Choices, 0 ) self.m_choice1.SetSelection( 0 ) bSizer31.Add( self.m_choice1, 0, wx.ALL, 5 ) bSizer6.Add( bSizer31, 1, wx.EXPAND, 5 ) #################################################### bSizer32 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText29 = wx.StaticText( self, wx.ID_ANY, u"框架类型 *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText29.Wrap( -1 ) bSizer32.Add( self.m_staticText29, 0, wx.ALL, 5 ) m_choice3Choices = [ u"无(0)", u"TensorFlow(1)", u"KalDi(2)", u"Caffe(3)"] self.m_choice3 = wx.Choice( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, m_choice3Choices, 0 ) self.m_choice3.SetSelection( 0 ) bSizer32.Add( self.m_choice3, 0, wx.ALL, 5 ) bSizer6.Add( bSizer32, 1, wx.EXPAND, 5 ) #################################################### bSizer33 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText30 = wx.StaticText( self, wx.ID_ANY, u"源数据(图片)路径 *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText30.Wrap( -1 ) bSizer33.Add( self.m_staticText30, 0, wx.ALL, 5 ) self.m_textCtrl27 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 500,-1 ), 0 ) bSizer33.Add( self.m_textCtrl27, 0, wx.ALL, 5 ) bSizer6.Add( bSizer33, 1, wx.EXPAND, 5 ) #################################################### bSizer45 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText31 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸N *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText31.Wrap( -1 ) bSizer45.Add( self.m_staticText31, 0, wx.ALL, 5 ) self.m_textCtrl28 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer45.Add( self.m_textCtrl28, 0, wx.ALL, 5 ) self.m_staticText32 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸C *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText32.Wrap( -1 ) bSizer45.Add( self.m_staticText32, 0, wx.ALL, 5 ) self.m_textCtrl29 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer45.Add( self.m_textCtrl29, 0, wx.ALL, 5 ) self.m_staticText33 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸H *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText33.Wrap( -1 ) bSizer45.Add( self.m_staticText33, 0, wx.ALL, 5 ) self.m_textCtrl30 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer45.Add( self.m_textCtrl30, 0, wx.ALL, 5 ) self.m_staticText34 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸W *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText34.Wrap( -1 ) bSizer45.Add( self.m_staticText34, 0, wx.ALL, 5 ) self.m_textCtrl31 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer45.Add( self.m_textCtrl31, 0, wx.ALL, 5 ) bSizer6.Add( bSizer45, 1, wx.EXPAND, 5 ) #################################################### bSizer47 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText35 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸N *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText35.Wrap( -1 ) bSizer47.Add( self.m_staticText35, 0, wx.ALL, 5 ) self.m_textCtrl32 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer47.Add( self.m_textCtrl32, 0, wx.ALL, 5 ) self.m_staticText36 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸C *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText36.Wrap( -1 ) bSizer47.Add( self.m_staticText36, 0, wx.ALL, 5 ) self.m_textCtrl33 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer47.Add( self.m_textCtrl33, 0, wx.ALL, 5 ) self.m_staticText37 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸H *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText37.Wrap( -1 ) bSizer47.Add( self.m_staticText37, 0, wx.ALL, 5 ) self.m_textCtrl34 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer47.Add( self.m_textCtrl34, 0, wx.ALL, 5 ) self.m_staticText38 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸W *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText38.Wrap( -1 ) bSizer47.Add( self.m_staticText38, 0, wx.ALL, 5 ) self.m_textCtrl35 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 ) bSizer47.Add( self.m_textCtrl35, 0, wx.ALL, 5 ) bSizer6.Add( bSizer47, 1, wx.EXPAND, 5 ) #################################################### bSizer48 = wx.BoxSizer( wx.HORIZONTAL ) self.m_staticText39 = wx.StaticText( self, wx.ID_ANY, u"超时时限 *", wx.DefaultPosition, wx.DefaultSize, 0 ) self.m_staticText39.Wrap( -1 ) bSizer48.Add( self.m_staticText39, 0, wx.ALL, 5 ) self.m_textCtrl36 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, 0 ) bSizer48.Add( self.m_textCtrl36, 0, wx.ALL, 5 ) bSizer6.Add( bSizer48, 1, wx.EXPAND, 5 ) #################################################### bSizer49 = wx.BoxSizer( wx.VERTICAL ) self.m_button2 = wx.Button( self, wx.ID_ANY, u"模型执行-异步-非级联", wx.DefaultPosition, wx.DefaultSize, 0 ) bSizer49.Add( self.m_button2, 0, wx.ALL, 5 ) bSizer6.Add( bSizer49, 1, wx.EXPAND, 5 ) self.Bind(wx.EVT_BUTTON, self.on_async_single, self.m_button2) #################################################### self.SetSizer( bSizer6 ) self.Layout() self.Centre( wx.BOTH ) ############################################################################################################################################################################### if self.m_checkBox2.GetValue() == True: self.m_textCtrl19.Enable() else: self.m_textCtrl19.Disable() if self.m_checkBox3.GetValue() == True: self.m_textCtrl21.Enable() else: self.m_textCtrl21.Disable() def on_async_single(self, event): model1Path = modelmngr_execute._getTextCtrl(self.m_textCtrl6.GetValue()) #1 model1Encr = modelmngr_execute._getCheckBox(self.m_checkBox2.GetValue()) #2 model1Key = modelmngr_execute._getTextCtrl(self.m_textCtrl19.GetValue()) #3 print model1Path, model1Encr, model1Key model2Path = modelmngr_execute._getTextCtrl(self.m_textCtrl20.GetValue()) #4 model2Encr = modelmngr_execute._getCheckBox(self.m_checkBox3.GetValue()) #5 model2Key = modelmngr_execute._getTextCtrl(self.m_textCtrl21.GetValue()) #6 print model2Path, model2Encr, model2Key modelType = str(self.m_choice1.GetSelection()) #7 frameworkType = str(self.m_choice3.GetSelection()) #8 jpgPath = modelmngr_execute._getTextCtrl(self.m_textCtrl27.GetValue()) #9 print modelType,frameworkType,jpgPath srcN = str(self.m_textCtrl28.GetValue()) #10 srcC = str(self.m_textCtrl29.GetValue()) #11 srcH = str(self.m_textCtrl30.GetValue()) #12 srcW = str(self.m_textCtrl31.GetValue()) #13 print srcN,srcC,srcH,srcW dstN = str(self.m_textCtrl32.GetValue()) #14 dstC = str(self.m_textCtrl33.GetValue()) #15 dstH = str(self.m_textCtrl34.GetValue()) #16 dstW = str(self.m_textCtrl35.GetValue()) #17 print dstN,dstC,dstH,dstW timeout = str(self.m_textCtrl36.GetValue()) #18 print timeout modelmngr_execute.singleRun_async(model1Path, model1Encr, model1Key, model2Path, model2Encr, model2Key, modelType, frameworkType, jpgPath, srcN, srcC, srcH, srcW, dstN, dstC, dstH, dstW, timeout) event.Skip() def on_model1_check(self, event): cb = event.GetEventObject() if cb.GetValue() == True: self.m_textCtrl19.Enable() self.m_textCtrl19.SetValue("") else: self.m_textCtrl19.Disable() self.m_textCtrl19.SetValue("") event.Skip() def on_model2_check(self, event): cb = event.GetEventObject() if cb.GetValue() == True: self.m_textCtrl21.Enable() self.m_textCtrl21.SetValue("") else: self.m_textCtrl21.Disable() self.m_textCtrl21.SetValue("") event.Skip() def __del__( self ): pass
时间: 2024-10-05 08:48:06