From 1981da9b22624789c780766c89d1fa4b0ad52584 Mon Sep 17 00:00:00 2001 From: Mark Shilton Date: Fri, 12 Feb 2016 13:16:29 +0000 Subject: [PATCH 1/2] Filename of current image is used as default name when saving label data --- labelImg.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/labelImg.py b/labelImg.py index c6c8db80..d9561778 100755 --- a/labelImg.py +++ b/labelImg.py @@ -920,17 +920,16 @@ class MainWindow(QMainWindow, WindowMixin): caption = '%s - Choose File' % __appname__ filters = 'File (*%s)' % LabelFile.suffix openDialogPath = self.currentPath() - dlg = QFileDialog(self, caption, openDialogPath, filters) + dlg = (self, caption, openDialogPath, filters) dlg.setDefaultSuffix(LabelFile.suffix[1:]) dlg.setAcceptMode(QFileDialog.AcceptSave) dlg.setConfirmOverwrite(True) + filenameWithoutExtension = os.path.splitext(self.filename)[0] + dlg.selectFile(filenameWithoutExtension) dlg.setOption(QFileDialog.DontUseNativeDialog, False) if dlg.exec_(): return dlg.selectedFiles()[0] return '' - #return unicode(QFileDialog.getSaveFileName(self, - # '%s - Choose File', self.currentPath(), - # 'Label files (*%s)' % LabelFile.suffix)) def _saveFile(self, filename): if filename and self.saveLabels(filename): @@ -1093,4 +1092,3 @@ def main(argv): if __name__ == '__main__': sys.exit(main(sys.argv)) - From 28062bafc476e8e699073a399d409250b7f4d9f0 Mon Sep 17 00:00:00 2001 From: Mark Shilton Date: Fri, 12 Feb 2016 13:23:12 +0000 Subject: [PATCH 2/2] Filename of current image is used as default name when saving label data --- labelImg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/labelImg.py b/labelImg.py index d9561778..e926866c 100755 --- a/labelImg.py +++ b/labelImg.py @@ -920,7 +920,7 @@ class MainWindow(QMainWindow, WindowMixin): caption = '%s - Choose File' % __appname__ filters = 'File (*%s)' % LabelFile.suffix openDialogPath = self.currentPath() - dlg = (self, caption, openDialogPath, filters) + dlg = QFileDialog(self, caption, openDialogPath, filters) dlg.setDefaultSuffix(LabelFile.suffix[1:]) dlg.setAcceptMode(QFileDialog.AcceptSave) dlg.setConfirmOverwrite(True)