delte alll modes except for YOLO_OBB

heshameraqi · Sep 2, 2019 · cdedadf · cdedadf
1 parent 3bb0e09
commit cdedadf
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/labelImg.py b/labelImg.py
@@ -502,29 +502,31 @@ def keyPressEvent(self, event):
 
     ## Support Functions ##
     def set_format(self, save_format):
+        '''
         if save_format == FORMAT_PASCALVOC:
             self.actions.save_format.setText(FORMAT_PASCALVOC)
             self.actions.save_format.setIcon(newIcon("format_voc"))
-            self.usingPascalVocFormat = True
+            self.usingPascalVocFormat = False
             self.usingYoloFormat = False
-            self.usingYoloOBBFormat = False
+            self.usingYoloOBBFormat = True
             LabelFile.suffix = XML_EXT
 
         elif save_format == FORMAT_YOLO:
             self.actions.save_format.setText(FORMAT_YOLO)
             self.actions.save_format.setIcon(newIcon("format_yolo"))
             self.usingPascalVocFormat = False
-            self.usingYoloFormat = True
-            self.usingYoloOBBFormat = False
-            LabelFile.suffix = TXT_EXT
-
-        elif save_format == FORMAT_YOLO_OBB:
-            self.actions.save_format.setText(FORMAT_YOLO_OBB)
-            self.actions.save_format.setIcon(newIcon("format_yolo_obb"))
-            self.usingPascalVocFormat = False
             self.usingYoloFormat = False
             self.usingYoloOBBFormat = True
             LabelFile.suffix = TXT_EXT
+            
+        elif save_format == FORMAT_YOLO_OBB:'''
+
+        self.actions.save_format.setText(FORMAT_YOLO_OBB)
+        self.actions.save_format.setIcon(newIcon("format_yolo_obb"))
+        self.usingPascalVocFormat = False
+        self.usingYoloFormat = False
+        self.usingYoloOBBFormat = True
+        LabelFile.suffix = TXT_EXT
 
     def change_format(self):
         if self.usingPascalVocFormat: self.set_format(FORMAT_YOLO)