From 777d03e25d3d0f768eac4a73fd02b3908d607020 Mon Sep 17 00:00:00 2001
From: aaronshan <shanruifeng@gmail.com>
Date: Wed, 17 May 2017 16:09:53 +0800
Subject: [PATCH] fix bugs:some pictures become invalid after repeat labeled
 and save it.

---
 labelImg.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/labelImg.py b/labelImg.py
index 8e6553fd..bfe46f53 100755
--- a/labelImg.py
+++ b/labelImg.py
@@ -1099,7 +1099,11 @@ class MainWindow(QMainWindow, WindowMixin):
                 savedPath = os.path.join(ustr(self.defaultSaveDir), savedFileName)
                 self._saveFile(savedPath)
         else:
-            self._saveFile(self.filePath if self.labelFile
+            imgFileDir = os.path.dirname(self.filePath)
+            imgFileName = os.path.basename(self.filePath)
+            savedFileName = os.path.splitext(imgFileName)[0] + LabelFile.suffix
+            savedPath = os.path.join(imgFileDir, savedFileName)
+            self._saveFile(savedPath if self.labelFile
                            else self.saveFileDialog())
 
     def saveFileAs(self, _value=False):