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

Merge pull request #8 from markd666/master

Browse Source 
Set a default file name when saving label data.
This commit is contained in:
darrenl 2016-02-15 19:24:26 +08:00
commit f6663a6327
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
labelImg.py
View File

@ -920,17 +920,16 @@ class MainWindow(QMainWindow, WindowMixin):
caption = '%s - Choose File' % __appname__ caption = '%s - Choose File' % __appname__
filters = 'File (*%s)' % LabelFile.suffix filters = 'File (*%s)' % LabelFile.suffix
openDialogPath = self.currentPath() openDialogPath = self.currentPath()
dlg = QFileDialog(self, caption, openDialogPath, filters) dlg = QFileDialog(self, caption, openDialogPath, filters)
dlg.setDefaultSuffix(LabelFile.suffix[1:]) dlg.setDefaultSuffix(LabelFile.suffix[1:])
dlg.setAcceptMode(QFileDialog.AcceptSave) dlg.setAcceptMode(QFileDialog.AcceptSave)
dlg.setConfirmOverwrite(True) dlg.setConfirmOverwrite(True)
filenameWithoutExtension = os.path.splitext(self.filename)[0]
dlg.selectFile(filenameWithoutExtension)
dlg.setOption(QFileDialog.DontUseNativeDialog, False) dlg.setOption(QFileDialog.DontUseNativeDialog, False)
if dlg.exec_(): if dlg.exec_():
return dlg.selectedFiles()[0] return dlg.selectedFiles()[0]
return '' return ''
#return unicode(QFileDialog.getSaveFileName(self,
# '%s - Choose File', self.currentPath(),
# 'Label files (*%s)' % LabelFile.suffix))
def _saveFile(self, filename): def _saveFile(self, filename):
if filename and self.saveLabels(filename): if filename and self.saveLabels(filename):
@ -1093,4 +1092,3 @@ def main(argv):
if __name__ == '__main__': if __name__ == '__main__':
sys.exit(main(sys.argv)) sys.exit(main(sys.argv))