Блог

У меня есть QPainterPath, и я хочу обрезать изображение, которое является QPixmap. Этот код сработал для меня, но я хочу использовать встроенную функциональность PyQt5, такую как mask, без numpy

 # read image as RGB and add alpha (transparency)
im = Image.open("frontal_1.jpg").convert("RGBA")

# convert to numpy (for convenience)
imArray = numpy.asarray(im)

# create mask
polygon = [(444, 203), (623, 243), (691, 177), (581, 26), (482, 42)]

maskIm = Image.new('L', (imArray.shape[1], imArray.shape[0]), 0)
ImageDraw.Draw(maskIm).polygon(polygon, outline=1, fill=1)
mask = numpy.array(maskIm)
...
newIm = Image.fromarray(newImArray, "RGBA")
newIm.save("out.png")
  

Один из возможных способов заменить маску — использовать метод setClipPath() QPainter:

 from PyQt5 import QtCore, QtGui

if __name__ == '__main__':
    image = QtGui.QImage('input.png')
    output = QtGui.QImage(image.size(), QtGui.QImage.Format_ARGB32)
    output.fill(QtCore.Qt.transparent)
    painter = QtGui.QPainter(output)

    points = [(444, 203), (623, 243), (691, 177), (581, 26), (482, 42)]
    polygon = QtGui.QPolygonF([QtCore.QPointF(*point) for point in points])

    path = QtGui.QPainterPath()
    path.addPolygon(polygon)
    painter.setClipPath(path)
    painter.drawImage(QtCore.QPoint(), image)
    painter.end()
    output.save('out.png')
  

После ответа выше я немного изменил свой код, и теперь он выглядит так:

     path = lips_contour_path
    image = QImage('frontal_2.jpg')
    output = QImage(image.size(), QImage.Format_ARGB32)
    output.fill(Qt.transparent)
    painter = QPainter(output)
    painter.setClipPath(path)
    painter.drawImage(QPoint(), image)
    painter.end()
    # To avoid useless transparent background you can crop it like that:
    output = output.copy(path.boundingRect().toRect())
    output.save('out.png')