diff --git a/data/predefined_classes.txt b/data/predefined_classes.txt new file mode 100644 index 00000000..94a82c14 --- /dev/null +++ b/data/predefined_classes.txt @@ -0,0 +1,5 @@ +dog +person +cat +tv +car diff --git a/labelImg.py b/labelImg.py index 8a0b4b31..621d4be7 100755 --- a/labelImg.py +++ b/labelImg.py @@ -72,11 +72,9 @@ class MainWindow(QMainWindow, WindowMixin): self.screencastViewer = "firefox" self.screencast = "https://youtu.be/p0nR2YsCY_U" + self.loadPredefinedClasses() # Main widgets and related state. - predefined_classes = ['person', 'dog'] - self.labelHist = predefined_classes self.labelDialog = LabelDialog(parent=self, listItem=self.labelHist) - self.labelList = QListWidget() self.itemsToShapes = {} self.shapesToItems = {} @@ -981,6 +979,16 @@ class MainWindow(QMainWindow, WindowMixin): self.canvas.endMove(copy=False) self.setDirty() + def loadPredefinedClasses(self): + predefined_classes_path = os.path.join('data', 'predefined_classes.txt') + if os.path.exists(predefined_classes_path) is True: + with open(predefined_classes_path) as f: + for line in f: + line = line.strip() + if self.labelHist is None: + self.lablHist = [line] + else: + self.labelHist.append(line) class Settings(object): """Convenience dict-like wrapper around QSettings."""