From 3e8da75cd0fdd1510146760fca6afbd5fb4faaf2 Mon Sep 17 00:00:00 2001 From: Yujia Hu Date: Mon, 12 Jan 2026 16:30:22 -0500 Subject: [PATCH 1/6] Update gui_categorizer.py --- LabGym/gui_categorizer.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/LabGym/gui_categorizer.py b/LabGym/gui_categorizer.py index 05a3f780..636ca0c0 100644 --- a/LabGym/gui_categorizer.py +++ b/LabGym/gui_categorizer.py @@ -1549,6 +1549,15 @@ 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() + self.text_selectcategorizer.SetLabel('The path to export the Categorizer: '+self.path_to_categorizer+'.') + dialog1.Destroy() + dialog.Destroy() + do_nothing=False stop=False From f2e0bd6fda99ed81e9ac0d0ded8ee84cb230be03 Mon Sep 17 00:00:00 2001 From: Yujia Hu Date: Mon, 12 Jan 2026 16:33:26 -0500 Subject: [PATCH 2/6] Update gui_detector.py --- LabGym/gui_detector.py | 8 ++++++++ 1 file changed, 8 insertions(+) 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 From 720c8e99abb95248466236c74e7378062f0d3d1a Mon Sep 17 00:00:00 2001 From: Yujia Hu Date: Mon, 12 Jan 2026 16:33:45 -0500 Subject: [PATCH 3/6] Update gui_categorizer.py --- LabGym/gui_categorizer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/LabGym/gui_categorizer.py b/LabGym/gui_categorizer.py index 636ca0c0..6a0585a9 100644 --- a/LabGym/gui_categorizer.py +++ b/LabGym/gui_categorizer.py @@ -1554,7 +1554,6 @@ def train_categorizer(self,event): dialog1=wx.DirDialog(self,'Select a directory','',style=wx.DD_DEFAULT_STYLE) if dialog1.ShowModal()==wx.ID_OK: self.path_to_categorizer=dialog1.GetPath() - self.text_selectcategorizer.SetLabel('The path to export the Categorizer: '+self.path_to_categorizer+'.') dialog1.Destroy() dialog.Destroy() From 55d6d7f2c69c34ab6a92d0a68b994ac15a136ca1 Mon Sep 17 00:00:00 2001 From: Yujia Hu Date: Mon, 12 Jan 2026 16:35:05 -0500 Subject: [PATCH 4/6] Update detector.py --- LabGym/detector.py | 1 + 1 file changed, 1 insertion(+) 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): From 9d01cb0e6e1d3d61e70934921d84ea59f9216861 Mon Sep 17 00:00:00 2001 From: Yujia Hu Date: Mon, 12 Jan 2026 16:42:25 -0500 Subject: [PATCH 5/6] Update gui_main.py --- LabGym/gui_main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) From 35fbe124b148eb71a6e4b208502885a54b51fc51 Mon Sep 17 00:00:00 2001 From: Yujia Hu Date: Mon, 12 Jan 2026 16:43:14 -0500 Subject: [PATCH 6/6] Update pyproject.toml --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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'},