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

Fix save dialog issue - #300

Browse Source
This commit is contained in:
vdalv 2018-05-15 23:12:35 -07:00 committed by darrenl
parent fef4e6b057
commit 508a4c8b37
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
labelImg.py
View File

@ -480,12 +480,14 @@ class MainWindow(QMainWindow, WindowMixin):
self.actions.save_format.setIcon(newIcon("format_voc")) self.actions.save_format.setIcon(newIcon("format_voc"))
self.usingPascalVocFormat = True self.usingPascalVocFormat = True
self.usingYoloFormat = False self.usingYoloFormat = False
LabelFile.suffix = XML_EXT
elif save_format == 'YOLO': elif save_format == 'YOLO':
self.actions.save_format.setText("YOLO") self.actions.save_format.setText("YOLO")
self.actions.save_format.setIcon(newIcon("format_yolo")) self.actions.save_format.setIcon(newIcon("format_yolo"))
self.usingPascalVocFormat = False self.usingPascalVocFormat = False
self.usingYoloFormat = True self.usingYoloFormat = True
LabelFile.suffix = TXT_EXT
def change_format(self): def change_format(self):
if self.usingPascalVocFormat: self.set_format("YOLO") if self.usingPascalVocFormat: self.set_format("YOLO")