Supported reading from Yolo and VOC formats then switching to YOLO_OBB

heshameraqi · Jan 9, 2019 · 88d80cb · 88d80cb
1 parent 39dea42
commit 88d80cb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/labelImg.py b/labelImg.py
@@ -1513,7 +1513,6 @@ def loadYOLOTXTByFilename(self, txtPath):
         shapes = tYoloParseReader.getShapes()
         print (shapes)
         self.loadLabels(shapes)
-        self.canvas.verified = tYoloParseReader.verified
 
     def loadYOLOTOBBXTByFilename(self, txtPath):
         if self.filePath is None:

diff --git a/libs/canvas.py b/libs/canvas.py
@@ -711,6 +711,8 @@ def loadPixmap(self, pixmap):
     def loadShapes(self, shapes):
         self.shapes = list(shapes)
         self.current = None
+        for s in shapes:
+            s.updateOBBInfo()
         self.repaint()
 
     def setShapeVisible(self, shape, value):

diff --git a/libs/shape.py b/libs/shape.py
@@ -136,7 +136,7 @@ def paint(self, painter):
                 min_y = min(min_y, point.y())
             if min_x != sys.maxsize and min_y != sys.maxsize:
                 font = QFont()
-                font.setPointSize(20)
+                font.setPointSize(10)
                 font.setBold(True)
                 painter.setFont(font)
                 if(self.label == None):