add truncated flag

Author: jjjkkkjjj-mizuno

labelImg.py

@@ -128,13 +128,27 @@ def __init__(self, defaultFilename=None, defaultPrefdefClassFile=None, defaultSa
         useDefaultLabelContainer = QWidget()
         useDefaultLabelContainer.setLayout(useDefaultLabelQHBoxLayout)
 
-        # Create a widget for edit and diffc button
-        self.diffcButton = QCheckBox(getStr('useDifficult'))
-        self.diffcButton.setChecked(False)
-        self.diffcButton.stateChanged.connect(self.btnstate)
         self.editButton = QToolButton()
         self.editButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
 
+        # Create Flag checkboxes list
+        self.flagGroupBox = QGroupBox(self)
+        self.flagGroupBox.setTitle('Flags')
+        flagslayout = QVBoxLayout(self)
+        self.flagGroupBox.setLayout(flagslayout)
+        self.flagButtons = []
+
+        self.diffcButton = QCheckBox(getStr('useDifficult'))
+        self.diffcButton.setChecked(False)
+        # calling stateChanged is inappropriate!
+        self.diffcButton.clicked.connect(self.btnstate)
+        self.flagButtons.append(self.diffcButton)
+
+        self.truncButton = QCheckBox(getStr('useTruncated'))
+        self.truncButton.setChecked(False)
+        self.truncButton.clicked.connect(self.btnstate)
+        self.flagButtons.append(self.truncButton)
+
         # *
         # * dhzs 2017-12-2 add copy button
         # *
@@ -143,9 +157,13 @@ def __init__(self, defaultFilename=None, defaultPrefdefClassFile=None, defaultSa
         self.copy_prev_button.clicked.connect(self.copyPreviousBoundingBoxes)
         listLayout.addWidget(self.copy_prev_button)
 
+        # add flag buttons to flagsGroupBox
+        for flagbtn in self.flagButtons:
+            flagslayout.addWidget(flagbtn)
+
         # Add some of widgets to listLayout
         listLayout.addWidget(self.editButton)
-        listLayout.addWidget(self.diffcButton)
+        listLayout.addWidget(self.flagGroupBox)
         listLayout.addWidget(useDefaultLabelContainer)
 
         # Create and add combobox for showing unique labels in group
@@ -446,8 +464,6 @@ def getFormatMeta(format):
         self.fillColor = None
         self.zoom_level = 100
         self.fit_window = False
-        # Add Chris
-        self.difficult = False
 
         ## Fix the compatible issue for qt4 and qt5. Convert the QStringList to python list
         if settings.get(SETTING_RECENT_FILES):
@@ -479,8 +495,7 @@ def getFormatMeta(format):
         Shape.line_color = self.lineColor = QColor(settings.get(SETTING_LINE_COLOR, DEFAULT_LINE_COLOR))
         Shape.fill_color = self.fillColor = QColor(settings.get(SETTING_FILL_COLOR, DEFAULT_FILL_COLOR))
         self.canvas.setDrawingColor(self.lineColor)
-        # Add chris
-        Shape.difficult = self.difficult
+        Shape.flags = self.flags
 
         def xbool(x):
             if isinstance(x, QVariant):
@@ -730,6 +745,20 @@ def fileitemDoubleClicked(self, item=None):
             if filename:
                 self.loadFile(filename)
 
+    @property
+    def flags(self):
+        """
+        Returns
+        -------
+        dict
+            key is a flag name
+            value is a Bool
+        """
+        return {flagbtn.text(): flagbtn.isChecked() for flagbtn in self.flagButtons}
+
+    def setFlagsChecked(self, flags):
+        pass
+
     # Add chris
     def btnstate(self, item= None):
         """ Function to handle difficult examples
@@ -741,16 +770,15 @@ def btnstate(self, item= None):
         if not item: # If not selected Item, take the first one
             item = self.labelList.item(self.labelList.count()-1)
 
-        difficult = self.diffcButton.isChecked()
+        flags = self.flags
 
         try:
             shape = self.itemsToShapes[item]
         except:
             pass
         # Checked and Update
         try:
-            if difficult != shape.difficult:
-                shape.difficult = difficult
+            if shape.setChangedFlags(flags):
                 self.setDirty()
             else:  # User probably changed item visibility
                 self.canvas.setShapeVisible(shape, item.checkState() == Qt.Checked)
@@ -798,7 +826,7 @@ def remLabel(self, shape):
 
     def loadLabels(self, shapes):
         s = []
-        for label, points, line_color, fill_color, difficult in shapes:
+        for label, points, line_color, fill_color, flags in shapes:
             shape = Shape(label=label)
             for x, y in points:
 
@@ -808,7 +836,7 @@ def loadLabels(self, shapes):
                     self.setDirty()
 
                 shape.addPoint(QPointF(x, y))
-            shape.difficult = difficult
+            shape.flags = flags
             shape.close()
             s.append(shape)
 
@@ -848,11 +876,10 @@ def format_shape(s):
                         line_color=s.line_color.getRgb(),
                         fill_color=s.fill_color.getRgb(),
                         points=[(p.x(), p.y()) for p in s.points],
-                       # add chris
-                        difficult = s.difficult)
+                        flags=s.flags)
 
         shapes = [format_shape(shape) for shape in self.canvas.shapes]
-        # Can add differrent annotation formats here
+        # Can add different annotation formats here
         try:
             if self.labelFileFormat == LabelFileFormat.PASCAL_VOC:
                 if annotationFilePath[-4:].lower() != ".xml":
@@ -899,9 +926,11 @@ def labelSelectionChanged(self):
             self._noSelectionSlot = True
             self.canvas.selectShape(self.itemsToShapes[item])
             shape = self.itemsToShapes[item]
-            # Add Chris
-            self.diffcButton.setChecked(shape.difficult)
-
+            for flagbtn in self.flagButtons:
+                if hasattr(shape, flagbtn.text()):
+                    flagbtn.setChecked(getattr(shape, flagbtn.text()))
+                else:
+                    flagbtn.setChecked(False)
     def labelItemChanged(self, item):
         shape = self.itemsToShapes[item]
         label = item.text()
@@ -932,8 +961,8 @@ def newShape(self):
         else:
             text = self.defaultLabelTextLine.text()
 
-        # Add Chris
-        self.diffcButton.setChecked(False)
+        for flagbtn in self.flagButtons:
+            flagbtn.setChecked(False)
         if text is not None:
             self.prevLabelText = text
             generate_color = generateColorByText(text)
@@ -1283,7 +1312,6 @@ def importDirImages(self, dirpath):
         self.lastOpenDir = dirpath
         self.dirname = dirpath
         self.filePath = None
-        self.fileListWidget.clear()
         self.mImgList = self.scanAllImages(dirpath)
         self.openNextImg()
         for imgPath in self.mImgList:

labelImg.spec

@@ -0,0 +1,33 @@
+# -*- mode: python ; coding: utf-8 -*-
+
+block_cipher = None
+
+
+a = Analysis(['labelImg.py'],
+             pathex=['./libs', './', 'C:\\Users\\Administrator\\Anaconda3\\envs\\labelimg\\Lib\\site-packages\\PyQt5\\Qt\\bin', 'O:\\Documents\\programs\\software\\labelImg_copyfunc'],
+             binaries=[],
+             datas=[],
+             hiddenimports=['xml', 'xml.etree', 'xml.etree.ElementTree', 'lxml.etree', 'PyQt5.sip'],
+             hookspath=[],
+             runtime_hooks=[],
+             excludes=[],
+             win_no_prefer_redirects=False,
+             win_private_assemblies=False,
+             cipher=block_cipher,
+             noarchive=False)
+pyz = PYZ(a.pure, a.zipped_data,
+             cipher=block_cipher)
+exe = EXE(pyz,
+          a.scripts,
+          a.binaries,
+          a.zipfiles,
+          a.datas,
+          [],
+          name='labelImg',
+          debug=False,
+          bootloader_ignore_signals=False,
+          strip=False,
+          upx=True,
+          upx_exclude=[],
+          runtime_tmpdir=None,
+          console=True )

libs/create_ml_io.py

@@ -125,7 +125,7 @@ def add_shape(self, label, bndbox):
         ymax = bndbox["y"] + (bndbox["height"] / 2)
 
         points = [(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax)]
-        self.shapes.append((label, points, None, None, True))
+        self.shapes.append((label, points, None, None, {'difficult': False, 'truncated': False}))
 
     def get_shapes(self):
         return self.shapes

libs/labelFile.py

@@ -77,10 +77,10 @@ def savePascalVocFormat(self, filename, shapes, imagePath, imageData,
         for shape in shapes:
             points = shape['points']
             label = shape['label']
-            # Add Chris
-            difficult = int(shape['difficult'])
+
+            flags = shape['flags']
             bndbox = LabelFile.convertPoints2BndBox(points)
-            writer.addBndBox(bndbox[0], bndbox[1], bndbox[2], bndbox[3], label, difficult)
+            writer.addBndBox(bndbox[0], bndbox[1], bndbox[2], bndbox[3], label, flags)
 
         writer.save(targetFile=filename)
         return
@@ -107,10 +107,10 @@ def saveYoloFormat(self, filename, shapes, imagePath, imageData, classList,
         for shape in shapes:
             points = shape['points']
             label = shape['label']
-            # Add Chris
-            difficult = int(shape['difficult'])
+
+            flags = shape['flags']
             bndbox = LabelFile.convertPoints2BndBox(points)
-            writer.addBndBox(bndbox[0], bndbox[1], bndbox[2], bndbox[3], label, difficult)
+            writer.addBndBox(bndbox[0], bndbox[1], bndbox[2], bndbox[3], label, flags)
 
         writer.save(targetFile=filename, classList=classList)
         return

libs/pascal_voc_io.py

@@ -77,10 +77,10 @@ def genXML(self):
         segmented.text = '0'
         return top
 
-    def addBndBox(self, xmin, ymin, xmax, ymax, name, difficult):
+    def addBndBox(self, xmin, ymin, xmax, ymax, name, flags):
         bndbox = {'xmin': xmin, 'ymin': ymin, 'xmax': xmax, 'ymax': ymax}
         bndbox['name'] = name
-        bndbox['difficult'] = difficult
+        bndbox['flags'] = flags
         self.boxlist.append(bndbox)
 
     def appendObjects(self, top):
@@ -90,15 +90,11 @@ def appendObjects(self, top):
             name.text = ustr(each_object['name'])
             pose = SubElement(object_item, 'pose')
             pose.text = "Unspecified"
-            truncated = SubElement(object_item, 'truncated')
-            if int(float(each_object['ymax'])) == int(float(self.imgSize[0])) or (int(float(each_object['ymin']))== 1):
-                truncated.text = "1" # max == height or min
-            elif (int(float(each_object['xmax']))==int(float(self.imgSize[1]))) or (int(float(each_object['xmin']))== 1):
-                truncated.text = "1" # max == width or min
-            else:
-                truncated.text = "0"
-            difficult = SubElement(object_item, 'difficult')
-            difficult.text = str( bool(each_object['difficult']) & 1 )
+
+            for key, val in each_object['flags'].items():
+                element = SubElement(object_item, key)
+                element.text = str( bool(val) & 1 )
+
             bndbox = SubElement(object_item, 'bndbox')
             xmin = SubElement(bndbox, 'xmin')
             xmin.text = str(each_object['xmin'])
@@ -128,7 +124,7 @@ class PascalVocReader:
 
     def __init__(self, filepath):
         # shapes type:
-        # [labbel, [(x1,y1), (x2,y2), (x3,y3), (x4,y4)], color, color, difficult]
+        # [labbel, [(x1,y1), (x2,y2), (x3,y3), (x4,y4)], color, color, flags]
         self.shapes = []
         self.filepath = filepath
         self.verified = False
@@ -140,13 +136,13 @@ def __init__(self, filepath):
     def getShapes(self):
         return self.shapes
 
-    def addShape(self, label, bndbox, difficult):
+    def addShape(self, label, bndbox, flags):
         xmin = int(float(bndbox.find('xmin').text))
         ymin = int(float(bndbox.find('ymin').text))
         xmax = int(float(bndbox.find('xmax').text))
         ymax = int(float(bndbox.find('ymax').text))
         points = [(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax)]
-        self.shapes.append((label, points, None, None, difficult))
+        self.shapes.append((label, points, None, None, flags))
 
     def parseXML(self):
         assert self.filepath.endswith(XML_EXT), "Unsupport file format"
@@ -163,9 +159,12 @@ def parseXML(self):
         for object_iter in xmltree.findall('object'):
             bndbox = object_iter.find("bndbox")
             label = object_iter.find('name').text
-            # Add chris
+
             difficult = False
             if object_iter.find('difficult') is not None:
                 difficult = bool(int(object_iter.find('difficult').text))
-            self.addShape(label, bndbox, difficult)
+            truncated = False
+            if object_iter.find('truncated') is not None:
+                truncated = bool(int(object_iter.find('truncated').text))
+            self.addShape(label, bndbox, flags={'difficult': difficult, 'truncated': truncated})
         return True

libs/shape.py

@@ -38,12 +38,12 @@ class Shape(object):
     scale = 1.0
     labelFontSize = 8
 
-    def __init__(self, label=None, line_color=None, difficult=False, paintLabel=False):
+    def __init__(self, label=None, line_color=None, flags={'difficult': False, 'truncated': False}, paintLabel=False):
         self.label = label
         self.points = []
         self.fill = False
         self.selected = False
-        self.difficult = difficult
+        self.flags = flags
         self.paintLabel = paintLabel
 
         self._highlightIndex = None
@@ -193,9 +193,30 @@ def copy(self):
             shape.line_color = self.line_color
         if self.fill_color != Shape.fill_color:
             shape.fill_color = self.fill_color
-        shape.difficult = self.difficult
+        shape.flags = self.flags
         return shape
 
+
+    def setChangedFlags(self, newflags):
+        """
+        Set changed flags and return the boolean representing whether to have changed flags
+        Returns
+        -------
+        Bool
+            Whether to have changed flags
+        """
+        isChanged = False
+        for newkey, newval in newflags.items():
+            if newkey not in self.flags.keys() or self.flags[newkey] != newval:
+                isChanged = True
+                break
+
+        if isChanged:
+            self.flags = newflags
+
+        return isChanged
+
+
     def __len__(self):
         return len(self.points)
 
@@ -204,3 +225,19 @@ def __getitem__(self, key):
 
     def __setitem__(self, key, value):
         self.points[key] = value
+
+    # allow to access the flag's key as Shape's attribute
+    def __getattr__(self, item):
+        if item in self.flags.keys():
+            return self.flags[item]
+        else:
+            raise AttributeError('Shape has no attribute \'{}\''.format(item))
+
+    def __setattr__(self, key, value):
+        try:
+            super().__setattr__(key, value)
+        except AttributeError:
+            if key in self.flags.keys():
+                self.flags[key] = value
+            else:
+                raise AttributeError('Shape has no attribute \'{}\''.format(key))