Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions LabGym/detector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
8 changes: 8 additions & 0 deletions LabGym/gui_categorizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
8 changes: 8 additions & 0 deletions LabGym/gui_detector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion LabGym/gui_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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'},
Expand Down