From 36a7d01e764d54c3c8d230d20a69042be12390ca Mon Sep 17 00:00:00 2001
From: Kentaro Wada <www.kentaro.wada@gmail.com>
Date: Wed, 11 May 2016 21:41:57 +0900
Subject: [PATCH] Set default filename with basename of image

---
 labelme/app.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/labelme/app.py b/labelme/app.py
index fa82ae6..2b62414 100644
--- a/labelme/app.py
+++ b/labelme/app.py
@@ -776,12 +776,13 @@ class MainWindow(QMainWindow, WindowMixin):
         dlg.setAcceptMode(QFileDialog.AcceptSave)
         dlg.setConfirmOverwrite(True)
         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))
+        basename = os.path.splitext(self.filename)[0]
+        default_labelfile_name = os.path.join(self.currentPath(),
+                                              basename + LabelFile.suffix)
+        filename = dlg.getSaveFileName(
+            self, 'Choose File', default_labelfile_name,
+            'Label files (*%s)' % LabelFile.suffix)
+        return unicode(filename)
 
     def _saveFile(self, filename):
         if filename and self.saveLabels(filename):
-- 
GitLab