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

Replace Difficult with difficult

Browse Source
This commit is contained in:
tzutalin 2017-05-10 21:26:51 +08:00
parent 8df94a45c8
commit a7e3a32385
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
labelImg.py Normal file → Executable file
View File

@ -132,7 +132,7 @@ class MainWindow(QMainWindow, WindowMixin):
self.editButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon) self.editButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
# Add chris # Add chris
self.diffcButton = QCheckBox("Difficult") self.diffcButton = QCheckBox("difficult")
self.diffcButton.setChecked(False) self.diffcButton.setChecked(False)
self.diffcButton.stateChanged.connect(self.btnstate) self.diffcButton.stateChanged.connect(self.btnstate)
@ -434,7 +434,7 @@ class MainWindow(QMainWindow, WindowMixin):
Shape.line_color = self.lineColor Shape.line_color = self.lineColor
Shape.fill_color = self.fillColor Shape.fill_color = self.fillColor
# Add chris # Add chris
Shape.Difficult = self.difficult Shape.difficult = self.difficult
def xbool(x): def xbool(x):
if isinstance(x, QVariant): if isinstance(x, QVariant):

4
libs/pascal_voc_io.py
View File

@ -94,7 +94,7 @@ class PascalVocWriter:
truncated.text = "1" # max == width or min truncated.text = "1" # max == width or min
else: else:
truncated.text = "0" truncated.text = "0"
difficult = SubElement(object_item, 'Difficult') difficult = SubElement(object_item, 'difficult')
difficult.text = str( bool(each_object['difficult']) & 1 ) difficult.text = str( bool(each_object['difficult']) & 1 )
bndbox = SubElement(object_item, 'bndbox') bndbox = SubElement(object_item, 'bndbox')
xmin = SubElement(bndbox, 'xmin') xmin = SubElement(bndbox, 'xmin')
@ -158,6 +158,6 @@ class PascalVocReader:
bndbox = object_iter.find("bndbox") bndbox = object_iter.find("bndbox")
label = object_iter.find('name').text label = object_iter.find('name').text
# Add chris # Add chris
difficult = bool(int(object_iter.find('Difficult').text)) difficult = bool(int(object_iter.find('difficult').text))
self.addShape(label, bndbox, difficult) self.addShape(label, bndbox, difficult)
return True return True