def _get_brush(self, color):
""" Returns a brush for the color.
"""
result = self._brushes.get(color)
if result is None:
qcolor = self._get_color(color)
result = QtGui.QBrush(qcolor)
self._brushes[color] = result
return result
def drawBboxes(self, qp, ignore=[]):
if self.image.isNull() or self.w == 0 or self.h == 0:
return
if not self.annotation:
return
# The overlay is created in the viewing coordinates
# This way, the drawing is more dense and the polygon edges are nicer
# We create an image that is the overlay
# Within this image we draw using another QPainter
# Finally we use the real QPainter to overlay the overlay-image on what is drawn so far
# The image that is used to draw the overlays
overlay = QtGui.QImage(self.w, self.h, QtGui.QImage.Format_ARGB32_Premultiplied)
# Fill the image
col = QtGui.QColor(0, 0, 0, 0)
overlay.fill(col)
# Create a new QPainter that draws in the overlay image
qp2 = QtGui.QPainter()
qp2.begin(overlay)
# Draw all objects
for obj in self.annotation.objects:
bbox, bboxVis = self.getBoundingBox(obj)
bboxToDraw = self.scaleBoundingBox(bbox)
bboxVisToDraw = self.scaleBoundingBox(bbox)
# The label of the object
name = obj.label
# If we do not know a color for this label, warn the user
if name not in name2labelCp:
print(
"The annotations contain unknown labels. This should not happen. Please inform the datasets authors. Thank you!")
print("Details: label '{}', file '{}'".format(name, self.currentLabelFile))
continue
# Reset brush for QPainter object
qp2.setBrush(QtGui.QBrush())
# Color from color table
col = QtGui.QColor(*name2labelCp[name].color)
if name2labelCp[name].hasInstances:
if self.highlightObj and obj == self.highlightObj:
pen = QtGui.QPen(QtGui.QBrush(col), 5.0)
else:
pen = QtGui.QPen(QtGui.QBrush(col), 3.0)
qp2.setPen(pen)
qp2.setOpacity(1.0)
qp2.drawRect(bboxToDraw)
if self.highlightObj and obj == self.highlightObj:
pen = QtGui.QPen(QtGui.QBrush(col), 3.0, style=QtCore.Qt.DotLine)
qp2.setPen(pen)
qp2.setOpacity(1.0)
qp2.drawRect(bboxVisToDraw)
else:
pen = QtGui.QPen(QtGui.QBrush(col), 1.0, style=QtCore.Qt.DashLine)
qp2.setPen(pen)
qp2.setOpacity(1.0)
qp2.drawRect(bboxVisToDraw)
qp2.setBrush(QtGui.QBrush(col, QtCore.Qt.SolidPattern))
qp2.setOpacity(0.4)
qp2.drawRect(bboxVisToDraw)
else:
if self.highlightObj and obj == self.highlightObj:
pen = QtGui.QPen(QtGui.QBrush(col), 3.0)
qp2.setPen(pen)
qp2.setBrush(QtGui.QBrush(col, QtCore.Qt.NoBrush))
else:
pen = QtGui.QPen(QtGui.QBrush(col), 1.0)
qp2.setPen(pen)
qp2.setBrush(QtGui.QBrush(col, QtCore.Qt.DiagCrossPattern))
qp2.setOpacity(1.0)
qp2.drawRect(bboxToDraw)
# End the drawing of the overlay
qp2.end()
# Save QPainter settings to stack
qp.save()
# Define transparency
qp.setOpacity(self.transp)
# Draw the overlay image
qp.drawImage(self.xoff, self.yoff, overlay)
# Restore settings
qp.restore()
return overlay
def drawLabels(self, qp, ignore=[]):
if self.image.isNull() or self.w == 0 or self.h == 0:
return
if not self.annotation:
return
# The overlay is created in the viewing coordinates
# This way, the drawing is more dense and the polygon edges are nicer
# We create an image that is the overlay
# Within this image we draw using another QPainter
# Finally we use the real QPainter to overlay the overlay-image on what is drawn so far
# The image that is used to draw the overlays
overlay = QtGui.QImage(self.w, self.h, QtGui.QImage.Format_ARGB32_Premultiplied)
# Fill the image with the default color
defaultLabel = name2label[self.defaultLabel]
col = QtGui.QColor(*defaultLabel.color)
overlay.fill(col)
# Create a new QPainter that draws in the overlay image
qp2 = QtGui.QPainter()
qp2.begin(overlay)
# The color of the outlines
qp2.setPen(QtGui.QColor('white'))
# Draw all objects
for obj in self.annotation.objects:
# The label of the object
name = assureSingleInstanceName(obj.label)
# If we do not know a color for this label, warn the user
if name not in name2label:
print(
"The annotations contain unkown labels. This should not happen. Please inform the datasets authors. Thank you!")
print("Details: label '{}', file '{}'".format(name, self.currentLabelFile))
continue
poly = self.getPolygon(obj)
# Scale the polygon properly
polyToDraw = poly * QtGui.QTransform.fromScale(self.scale, self.scale)
# Default drawing
# Color from color table, solid brush
col = QtGui.QColor(*name2label[name].color)
brush = QtGui.QBrush(col, QtCore.Qt.SolidPattern)
qp2.setBrush(brush)
# Overwrite drawing if this is the highlighted object
if self.highlightObj and obj == self.highlightObj:
# First clear everything below of the polygon
qp2.setCompositionMode(QtGui.QPainter.CompositionMode_Clear)
qp2.drawPolygon(polyToDraw)
qp2.setCompositionMode(QtGui.QPainter.CompositionMode_SourceOver)
# Set the drawing to a special pattern
brush = QtGui.QBrush(col, QtCore.Qt.DiagCrossPattern)
qp2.setBrush(brush)
qp2.drawPolygon(polyToDraw)
# Draw outline of selected object dotted
if self.highlightObj:
brush = QtGui.QBrush(QtCore.Qt.NoBrush)
qp2.setBrush(brush)
qp2.setPen(QtCore.Qt.DashLine)
polyToDraw = self.getPolygon(self.highlightObj) * QtGui.QTransform.fromScale(self.scale, self.scale)
qp2.drawPolygon(polyToDraw)
# End the drawing of the overlay
qp2.end()
# Save QPainter settings to stack
qp.save()
# Define transparency
qp.setOpacity(self.transp)
# Draw the overlay image
qp.drawImage(self.xoff, self.yoff, overlay)
# Restore settings
qp.restore()
return overlay
