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

Fix the wrong path/width/height in annotation files

Browse Source
This commit is contained in:
tzutalin 2017-02-24 22:11:38 +08:00
parent 700bba348f
commit 34420d4053
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
labelImg.py
View File

@ -638,7 +638,8 @@ class MainWindow(QMainWindow, WindowMixin):
shape.fill_color = QColor(*fill_color) shape.fill_color = QColor(*fill_color)
self.canvas.loadShapes(s) self.canvas.loadShapes(s)
def saveLabels(self, filePath): def saveLabels(self, annotationFilePath):
annotationFilePath = u(annotationFilePath)
lf = LabelFile() lf = LabelFile()
def format_shape(s): def format_shape(s):
return dict(label=s.label, return dict(label=s.label,
@ -651,15 +652,15 @@ class MainWindow(QMainWindow, WindowMixin):
shapes = [format_shape(shape) for shape in self.canvas.shapes] shapes = [format_shape(shape) for shape in self.canvas.shapes]
# Can add differrent annotation formats here # Can add differrent annotation formats here
try: try:
unicodeFilePath = u(filePath) annotationFilePath = u(annotationFilePath)
if self.usingPascalVocFormat is True: if self.usingPascalVocFormat is True:
lf.savePascalVocFormat(unicodeFilePath, shapes, unicodeFilePath, self.imageData, print ('Img: ' + self.filePath + ' Xml: ' + annotationFilePath)
lf.savePascalVocFormat(annotationFilePath, shapes, self.filePath, self.imageData,
self.lineColor.getRgb(), self.fillColor.getRgb()) self.lineColor.getRgb(), self.fillColor.getRgb())
else: else:
lf.save(unicodeFilePath, shapes, unicodeFilePath, self.imageData, lf.save(annotationFilePath, shapes, self.filePath, self.imageData,
self.lineColor.getRgb(), self.fillColor.getRgb()) self.lineColor.getRgb(), self.fillColor.getRgb())
self.labelFile = lf self.labelFile = lf
self.filePath = unicodeFilePath
return True return True
except LabelFileError as e: except LabelFileError as e:
self.errorMessage(u'Error saving label data', self.errorMessage(u'Error saving label data',