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

Save used labeling format in settings

Browse Source
This commit is contained in:
Laszlo Treszkai 2020-07-16 15:48:14 +02:00 committed by darrenl
parent b21b48e640
commit 196bd46828
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
labelImg.py
View File

@ -86,8 +86,8 @@ class MainWindow(QMainWindow, WindowMixin):
# Save as Pascal voc xml # Save as Pascal voc xml
self.defaultSaveDir = defaultSaveDir self.defaultSaveDir = defaultSaveDir
self.usingPascalVocFormat = True self.usingPascalVocFormat = settings.get(SETTING_USING_VOC_FORMAT, True)
self.usingYoloFormat = False self.usingYoloFormat = not self.usingPascalVocFormat
# For loading all image under a directory # For loading all image under a directory
self.mImgList = [] self.mImgList = []
@ -229,8 +229,10 @@ class MainWindow(QMainWindow, WindowMixin):
save = action(getStr('save'), self.saveFile, save = action(getStr('save'), self.saveFile,
'Ctrl+S', 'save', getStr('saveDetail'), enabled=False) 'Ctrl+S', 'save', getStr('saveDetail'), enabled=False)
save_format = action('&PascalVOC', self.change_format, save_format = action('&PascalVOC' if self.usingPascalVocFormat else '&YOLO',
'Ctrl+', 'format_voc', getStr('changeSaveFormat'), enabled=True) self.change_format, 'Ctrl+',
'format_voc' if self.usingPascalVocFormat else 'format_yolo',
getStr('changeSaveFormat'), enabled=True)
saveAs = action(getStr('saveAs'), self.saveFileAs, saveAs = action(getStr('saveAs'), self.saveFileAs,
'Ctrl+Shift+S', 'save-as', getStr('saveAsDetail'), enabled=False) 'Ctrl+Shift+S', 'save-as', getStr('saveAsDetail'), enabled=False)
@ -1155,6 +1157,7 @@ class MainWindow(QMainWindow, WindowMixin):
settings[SETTING_SINGLE_CLASS] = self.singleClassMode.isChecked() settings[SETTING_SINGLE_CLASS] = self.singleClassMode.isChecked()
settings[SETTING_PAINT_LABEL] = self.displayLabelOption.isChecked() settings[SETTING_PAINT_LABEL] = self.displayLabelOption.isChecked()
settings[SETTING_DRAW_SQUARE] = self.drawSquaresOption.isChecked() settings[SETTING_DRAW_SQUARE] = self.drawSquaresOption.isChecked()
settings[SETTING_USING_VOC_FORMAT] = self.usingPascalVocFormat
settings.save() settings.save()
def loadRecent(self, filename): def loadRecent(self, filename):

1
libs/constants.py
View File

@ -15,4 +15,5 @@ SETTING_SINGLE_CLASS = 'singleclass'
FORMAT_PASCALVOC='PascalVOC' FORMAT_PASCALVOC='PascalVOC'
FORMAT_YOLO='YOLO' FORMAT_YOLO='YOLO'
SETTING_DRAW_SQUARE = 'draw/square' SETTING_DRAW_SQUARE = 'draw/square'
SETTING_USING_VOC_FORMAT = 'usingVocFormat'
DEFAULT_ENCODING = 'utf-8' DEFAULT_ENCODING = 'utf-8'