Эх сурвалжийг харах

Overwrite existing label file with importDir

Kentaro Wada 7 жил өмнө
parent
commit
c7442b7735

+ 1 - 1
labelme/app.py

@@ -917,7 +917,7 @@ class MainWindow(QMainWindow, WindowMixin):
         if self.hasLabels():
             if self.labelFile:
                 # DL20180323 - overwrite when in directory
-                self._saveFile(self.saveFileDialog())
+                self._saveFile(self.labelFile.filename)
             elif self.output:
                 self._saveFile(self.output)
             else:

+ 3 - 0
labelme/labelFile.py

@@ -38,6 +38,7 @@ class LabelFile(object):
         self.imageData = None
         if filename is not None:
             self.load(filename)
+        self.filename = filename
 
     def load(self, filename):
         try:
@@ -61,6 +62,7 @@ class LabelFile(object):
                 self.imageData = imageData
                 self.lineColor = lineColor
                 self.fillColor = fillColor
+                self.filename = filename
         except Exception as e:
             raise LabelFileError(e)
 
@@ -78,6 +80,7 @@ class LabelFile(object):
         try:
             with open(filename, 'wb' if PY2 else 'w') as f:
                 json.dump(data, f, ensure_ascii=True, indent=2)
+            self.filename = filename
         except Exception as e:
             raise LabelFileError(e)