From 4a25352dbea633278557a9900125ceb6af9d0dcc Mon Sep 17 00:00:00 2001
From: RegisWANG <511545630@qq.com>
Date: Tue, 23 May 2017 23:27:31 +0800
Subject: [PATCH] Add use default label checkbox

---
 labelImg.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/labelImg.py b/labelImg.py
index beaf91d6..5ddbef3f 100755
--- a/labelImg.py
+++ b/labelImg.py
@@ -133,9 +133,17 @@ class MainWindow(QMainWindow, WindowMixin):
         self.diffcButton.setChecked(False)
         self.diffcButton.stateChanged.connect(self.btnstate)
 
+        #Add Use Default Checkbox
+        self.useDefautLabel = QCheckBox("use default label")
+        self.useDefautLabel.setChecked(False)
+        self.defaultLabel = QLineEdit()
+
         self.labelListContainer = QWidget()
         self.labelListContainer.setLayout(listLayout)
         listLayout.addWidget(self.editButton)  # , 0, Qt.AlignCenter)
+        listLayout.addWidget(self.useDefautLabel)
+        listLayout.addWidget(self.defaultLabel)
+
         # Add chris
         listLayout.addWidget(self.diffcButton)
         listLayout.addWidget(self.labelList)
@@ -753,11 +761,15 @@ class MainWindow(QMainWindow, WindowMixin):
 
         position MUST be in global coordinates.
         """
-        if len(self.labelHist) > 0:
-            self.labelDialog = LabelDialog(
-                parent=self, listItem=self.labelHist)
+        if not self.useDefautLabel.isChecked() or not self.defaultLabel.text():
+            if len(self.labelHist) > 0:
+                self.labelDialog = LabelDialog(
+                    parent=self, listItem=self.labelHist)
+
+            text = self.labelDialog.popUp(text=self.prevLabelText)
+        else:
+            text = self.defaultLabel.text()
 
-        text = self.labelDialog.popUp(text=self.prevLabelText)
         # Add Chris
         self.diffcButton.setChecked(False)
         if text is not None: