apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Save autosave/single class mode to setting file

Browse Source
This commit is contained in:
tzutalin 2017-10-14 00:01:13 +08:00
parent 1c5298f13d
commit 713cf4537e
2 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
labelImg.py
View File

@ -87,6 +87,12 @@ class MainWindow(QMainWindow, WindowMixin):
def __init__(self, defaultFilename=None, defaultPrefdefClassFile=None): def __init__(self, defaultFilename=None, defaultPrefdefClassFile=None):
super(MainWindow, self).__init__() super(MainWindow, self).__init__()
self.setWindowTitle(__appname__) self.setWindowTitle(__appname__)
# Load setting in the main thread
self.settings = Settings()
self.settings.load()
settings = self.settings
# Save as Pascal voc xml # Save as Pascal voc xml
self.defaultSaveDir = None self.defaultSaveDir = None
self.usingPascalVocFormat = True self.usingPascalVocFormat = True
@ -349,11 +355,12 @@ class MainWindow(QMainWindow, WindowMixin):
# Auto saving : Enable auto saving if pressing next # Auto saving : Enable auto saving if pressing next
self.autoSaving = QAction("Auto Saving", self) self.autoSaving = QAction("Auto Saving", self)
self.autoSaving.setCheckable(True) self.autoSaving.setCheckable(True)
self.autoSaving.setChecked(settings.get(SETTING_AUTO_SAVE, False))
# Sync single class mode from PR#106 # Sync single class mode from PR#106
self.singleClassMode = QAction("Single Class Mode", self) self.singleClassMode = QAction("Single Class Mode", self)
self.singleClassMode.setShortcut("Ctrl+Shift+S") self.singleClassMode.setShortcut("Ctrl+Shift+S")
self.singleClassMode.setCheckable(True) self.singleClassMode.setCheckable(True)
self.singleClassMode.setChecked(settings.get(SETTING_SINGLE_CLASS, False))
self.lastLabel = None self.lastLabel = None
addActions(self.menus.file, addActions(self.menus.file,
@ -400,10 +407,6 @@ class MainWindow(QMainWindow, WindowMixin):
# Add Chris # Add Chris
self.difficult = False self.difficult = False
self.settings = Settings()
self.settings.load()
settings = self.settings
## Fix the compatible issue for qt4 and qt5. Convert the QStringList to python list ## Fix the compatible issue for qt4 and qt5. Convert the QStringList to python list
if settings.get(SETTING_RECENT_FILES): if settings.get(SETTING_RECENT_FILES):
if have_qstring(): if have_qstring():
@ -1003,6 +1006,8 @@ class MainWindow(QMainWindow, WindowMixin):
else: else:
settings[SETTING_LAST_OPEN_DIR] = "" settings[SETTING_LAST_OPEN_DIR] = ""
settings[SETTING_AUTO_SAVE] = self.autoSaving.isChecked()
settings[SETTING_SINGLE_CLASS] = self.singleClassMode.isChecked()
settings.save() settings.save()
## User Dialogs ## ## User Dialogs ##

2
libs/constants.py
View File

@ -9,3 +9,5 @@ SETTING_ADVANCE_MODE = 'advanced'
SETTING_WIN_STATE = 'window/state' SETTING_WIN_STATE = 'window/state'
SETTING_SAVE_DIR = 'savedir' SETTING_SAVE_DIR = 'savedir'
SETTING_LAST_OPEN_DIR = 'lastOpenDir' SETTING_LAST_OPEN_DIR = 'lastOpenDir'
SETTING_AUTO_SAVE = 'autosave'
SETTING_SINGLE_CLASS = 'singleclass'