f5106a2a6a
LabelImg cannot read .lif files, but saving the annotations as .xml allows the annotations to be read when switching between images or reopening previously annotated images.
88 lines
2.6 KiB
Python
88 lines
2.6 KiB
Python
# Copyright (c) 2016 Tzutalin
|
|
# Create by TzuTaLin <tzu.ta.lin@gmail.com>
|
|
|
|
try:
|
|
from PyQt5.QtGui import QImage
|
|
except ImportError:
|
|
from PyQt4.QtGui import QImage
|
|
|
|
from base64 import b64encode, b64decode
|
|
from pascal_voc_io import PascalVocWriter
|
|
import os.path
|
|
import sys
|
|
|
|
|
|
class LabelFileError(Exception):
|
|
pass
|
|
|
|
|
|
class LabelFile(object):
|
|
# It might be changed as window creates
|
|
suffix = '.xml'
|
|
|
|
def __init__(self, filename=None):
|
|
self.shapes = ()
|
|
self.imagePath = None
|
|
self.imageData = None
|
|
self.verified = False
|
|
|
|
def savePascalVocFormat(self, filename, shapes, imagePath, imageData,
|
|
lineColor=None, fillColor=None, databaseSrc=None):
|
|
imgFolderPath = os.path.dirname(imagePath)
|
|
imgFolderName = os.path.split(imgFolderPath)[-1]
|
|
imgFileName = os.path.basename(imagePath)
|
|
imgFileNameWithoutExt = os.path.splitext(imgFileName)[0]
|
|
# Read from file path because self.imageData might be empty if saving to
|
|
# Pascal format
|
|
image = QImage()
|
|
image.load(imagePath)
|
|
imageShape = [image.height(), image.width(),
|
|
1 if image.isGrayscale() else 3]
|
|
writer = PascalVocWriter(imgFolderName, imgFileNameWithoutExt,
|
|
imageShape, localImgPath=imagePath)
|
|
writer.verified = self.verified
|
|
|
|
for shape in shapes:
|
|
points = shape['points']
|
|
label = shape['label']
|
|
# Add Chris
|
|
difficult = int(shape['difficult'])
|
|
bndbox = LabelFile.convertPoints2BndBox(points)
|
|
writer.addBndBox(bndbox[0], bndbox[1], bndbox[2], bndbox[3], label, difficult)
|
|
|
|
writer.save(targetFile=filename)
|
|
return
|
|
|
|
def toggleVerify(self):
|
|
self.verified = not self.verified
|
|
|
|
@staticmethod
|
|
def isLabelFile(filename):
|
|
fileSuffix = os.path.splitext(filename)[1].lower()
|
|
return fileSuffix == LabelFile.suffix
|
|
|
|
@staticmethod
|
|
def convertPoints2BndBox(points):
|
|
xmin = float('inf')
|
|
ymin = float('inf')
|
|
xmax = float('-inf')
|
|
ymax = float('-inf')
|
|
for p in points:
|
|
x = p[0]
|
|
y = p[1]
|
|
xmin = min(x, xmin)
|
|
ymin = min(y, ymin)
|
|
xmax = max(x, xmax)
|
|
ymax = max(y, ymax)
|
|
|
|
# Martin Kersner, 2015/11/12
|
|
# 0-valued coordinates of BB caused an error while
|
|
# training faster-rcnn object detector.
|
|
if xmin < 1:
|
|
xmin = 1
|
|
|
|
if ymin < 1:
|
|
ymin = 1
|
|
|
|
return (int(xmin), int(ymin), int(xmax), int(ymax))
|