如何将PIL.Image.Image类型图片添加至PyQt的QPushButton组件？

Got it, let's walk through how to get your PIL Image into a QPushButton. Qt widgets like QPushButton rely on its own image types (QPixmap/QIcon) instead of PIL's Image, so we just need to convert the format properly. Here's a step-by-step solution:

Step 1: Convert PIL Image to QImage #

First, we need to translate the PIL Image data into a QImage (Qt's native image format). We'll handle different image modes (like RGB/RGBA) to make sure the conversion works correctly:

from PIL import Image
from PyQt5.QtGui import QImage, QPixmap, QIcon
from PyQt5.QtWidgets import QPushButton, QApplication
import sys

# Your existing PIL Image object
im = w.to_image()  # Output of type <class 'PIL.Image.Image'>

# Get image dimensions and mode
width, height = im.size
image_mode = im.mode

# Map PIL modes to QImage formats
if image_mode == "RGB":
    qt_format = QImage.Format_RGB888
elif image_mode == "RGBA":
    qt_format = QImage.Format_RGBA8888
else:
    # Fallback: convert non-RGB/RGBA images to RGB (e.g., grayscale)
    im = im.convert("RGB")
    qt_format = QImage.Format_RGB888
    width, height = im.size

# Create QImage from PIL's byte data
q_image = QImage(im.tobytes(), width, height, qt_format)

Step 2: Convert QImage to QPixmap #

QPushButton works best with QPixmap (it's optimized for screen display). Convert the QImage to a QPixmap with one simple method:

pixmap = QPixmap.fromImage(q_image)

Step 3: Assign the Pixmap to the QPushButton #

Now set the pixmap as the button's icon, and adjust the icon size to fit your needs:

# Initialize the app and button
app = QApplication(sys.argv)
button = QPushButton()

# Set the icon and adjust its size
button.setIcon(QIcon(pixmap))
# Option 1: Match icon size to the pixmap's original size
button.setIconSize(pixmap.size())
# Option 2: Set a fixed size (e.g., 150x150) while keeping aspect ratio
# button.setIconSize(pixmap.scaled(150, 150, Qt.KeepAspectRatio).size())

button.show()
sys.exit(app.exec_())

Notes for Edge Cases #

• Large Images: If your PIL Image is too big, scale the pixmap before setting it to avoid stretching the button:

  scaled_pixmap = pixmap.scaled(button.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)
  button.setIcon(QIcon(scaled_pixmap))
  

• PySide Instead of PyQt: The code works almost identically—just replace PyQt5 with PySide2 or PySide6 in your imports.
• Transparent Images: Make sure to use the Format_RGBA8888 format if your image has an alpha channel, otherwise transparency will be lost.

内容的提问来源于stack exchange，提问作者Kobshobe