diff --git a/LabGym/detector.py b/LabGym/detector.py index d06a8a6f..a570653f 100644 --- a/LabGym/detector.py +++ b/LabGym/detector.py @@ -127,6 +127,7 @@ def train(self,path_to_annotation,path_to_trainingimages,path_to_detector,iterat f.write(cfg.dump()) print('Detector training completed!') + print('Trained Detector saved in: '+str(path_to_detector)) def test(self,path_to_annotation,path_to_testingimages,path_to_detector,output_path): diff --git a/LabGym/gui_categorizer.py b/LabGym/gui_categorizer.py index 05a3f780..6a0585a9 100644 --- a/LabGym/gui_categorizer.py +++ b/LabGym/gui_categorizer.py @@ -1549,6 +1549,14 @@ def train_categorizer(self,event): else: + dialog=wx.MessageDialog(self,'Export the trained Categorizer to a folder?','Export trained Categorizer?',wx.YES_NO|wx.ICON_QUESTION) + if dialog.ShowModal()==wx.ID_YES: + dialog1=wx.DirDialog(self,'Select a directory','',style=wx.DD_DEFAULT_STYLE) + if dialog1.ShowModal()==wx.ID_OK: + self.path_to_categorizer=dialog1.GetPath() + dialog1.Destroy() + dialog.Destroy() + do_nothing=False stop=False diff --git a/LabGym/gui_detector.py b/LabGym/gui_detector.py index 0e5e83bc..5cec30fb 100644 --- a/LabGym/gui_detector.py +++ b/LabGym/gui_detector.py @@ -359,6 +359,14 @@ def train_detector(self,event): else: + dialog=wx.MessageDialog(self,'Export the trained Detector to a folder?','Export trained Detector?',wx.YES_NO|wx.ICON_QUESTION) + if dialog.ShowModal()==wx.ID_YES: + dialog1=wx.DirDialog(self,'Select a directory','',style=wx.DD_DEFAULT_STYLE) + if dialog1.ShowModal()==wx.ID_OK: + self.path_to_detector=dialog1.GetPath() + dialog1.Destroy() + dialog.Destroy() + do_nothing=False stop=False diff --git a/LabGym/gui_main.py b/LabGym/gui_main.py index 8f32a4fc..cfd263dc 100644 --- a/LabGym/gui_main.py +++ b/LabGym/gui_main.py @@ -66,7 +66,7 @@ def display_window(self): boxsizer.Add(self.text_welcome,0,wx.LEFT|wx.RIGHT|wx.EXPAND,5) boxsizer.Add(0,60,0) self.text_developers=wx.StaticText(panel, - label='Created by Yujia Hu and Bing Ye\n\nLife Sciences Institute, University of Michigan\n\n\n\nContributor list:\n\nJie Zhou, Rohan Satapathy, John Ruckstuhl, Brendon O. Waston, Carrie R. Ferrario,\n\nKelly Goss, Isabelle Baker, M. Victor Struman, Bobby Tomlinson',style=wx.ALIGN_CENTER|wx.ST_ELLIPSIZE_END) + label='Created by Yujia Hu and Bing Ye\n\nLife Sciences Institute, University of Michigan\n\n\n\nContributor list:\n\nJie Zhou, John Ruckstuhl, Rohan Satapathy, Brendon O. Waston, Carrie R. Ferrario,\n\nKelly Goss, Isabelle Baker, M. Victor Struman, Bobby Tomlinson',style=wx.ALIGN_CENTER|wx.ST_ELLIPSIZE_END) boxsizer.Add(self.text_developers,0,wx.LEFT|wx.RIGHT|wx.EXPAND,5) boxsizer.Add(0,60,0) diff --git a/pyproject.toml b/pyproject.toml index c9bff2a0..0e68a53f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -7,8 +7,8 @@ name='LabGym' description='Quantify user-defined behaviors.' authors=[ {name='Yujia Hu',email='yujiahu415@gmail.com'}, - {name='Rohan Satapathy',email='rohansat@umich.edu'}, {name='John Ruckstuhl',email='john.ruckstuhl@gmail.com'}, + {name='Rohan Satapathy',email='rohansat@umich.edu'}, {name='M. Victor Struman',email='strmark@umich.edu'}, {name='Kelly Goss',email='khgoss@umich.edu'}, {name='Isabelle Baker',email='ibaker@umich.edu'},