apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add predefined class text file

Browse Source
This commit is contained in:
tzutalin 2015-11-21 00:15:21 +08:00
parent 66645103e5
commit aa1860b497
2 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
data/predefined_classes.txt Normal file
View File

@ -0,0 +1,5 @@
dog
person
cat
tv
car

14
labelImg.py
View File

@ -72,11 +72,9 @@ class MainWindow(QMainWindow, WindowMixin):
self.screencastViewer = "firefox" self.screencastViewer = "firefox"
self.screencast = "https://youtu.be/p0nR2YsCY_U" self.screencast = "https://youtu.be/p0nR2YsCY_U"
self.loadPredefinedClasses()
# Main widgets and related state. # Main widgets and related state.
predefined_classes = ['person', 'dog']
self.labelHist = predefined_classes
self.labelDialog = LabelDialog(parent=self, listItem=self.labelHist) self.labelDialog = LabelDialog(parent=self, listItem=self.labelHist)
self.labelList = QListWidget() self.labelList = QListWidget()
self.itemsToShapes = {} self.itemsToShapes = {}
self.shapesToItems = {} self.shapesToItems = {}
@ -981,6 +979,16 @@ class MainWindow(QMainWindow, WindowMixin):
self.canvas.endMove(copy=False) self.canvas.endMove(copy=False)
self.setDirty() 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): class Settings(object):
"""Convenience dict-like wrapper around QSettings.""" """Convenience dict-like wrapper around QSettings."""