Сбои Python при попытке просмотра EXR-файлов в PYQT5 - PullRequest
Новая
       56

Сбои Python при попытке просмотра EXR-файлов в PYQT5

0 голосов
/ 18 октября 2018

Я пытаюсь загрузить два exr-файла и загрузить их в метки на графическом интерфейсе, чтобы я мог просматривать два файла рядом.Я могу заставить один работать, но когда я пытаюсь загрузить оба python, вылетает.Ниже мой код: 

def exrToJpgGamma(exrfile):
    file = OpenEXR.InputFile(exrfile)
    pt = Imath.PixelType(Imath.PixelType.FLOAT)
    dw = file.header()['dataWindow']
    size = (dw.max.x - dw.min.x + 1, dw.max.y - dw.min.y + 1)

    RedStr = file.channel('R', pt)
    GreenStr = file.channel('G', pt)
    BlueStr = file.channel('B', pt)

    Red = array.array('f', RedStr)
    Green = array.array('f', GreenStr)
    Blue = array.array('f', BlueStr)

    def EncodeToSRGB(v):
        if (v <= 0.0031308):
            return (v * 12.92) * 255.0
        else:
            return (1.055*(v**(1.0/2.2))-0.055) * 255.0

    for I in range(len(Red)):
        Red[I] = EncodeToSRGB(Red[I])
    for I in range(len(Green)):
        Green[I] = EncodeToSRGB(Green[I])

    for I in range(len(Blue)):
        Blue[I] = EncodeToSRGB(Blue[I])

    rgbf = [Image.frombytes("F", size, Red.tobytes())]
    rgbf.append(Image.frombytes("F", size, Green.tobytes()))
    rgbf.append(Image.frombytes("F", size, Blue.tobytes()))

    rgb8 = [im.convert("L") for im in rgbf]
    myqimage = Image.merge("RGB", rgb8)
    return myqimage

def showEXR(self):
    width = 480
    height = 360

    imageq = PilImageQt(exrToJpgGamma(chip.exr))
    qimage = QtGui.QImage(imageq)
    pixmap = QtGui.QPixmap.fromImage(qimage)
    ScaledPixmap = pixmap.scaled(width, height, QtCore.Qt.KeepAspectRatio, QtCore.Qt.FastTransformation)
    self.chip_img.setPixmap(ScaledPixmap)

    imageq = PilImageQt(exrToJpgGamma(panel.exr))
    qimage = QtGui.QImage(imageq)
    pixmap = QtGui.QPixmap.fromImage(qimage)
    ScaledPixmap = pixmap.scaled(width, height, QtCore.Qt.KeepAspectRatio, QtCore.Qt.FastTransformation)
    self.panel_img.setPixmap(ScaledPixmap)

    return

showEXR(self)

Дайте мне знать, если вам нужны какие-либо дополнительные детали.Заранее спасибо.

Вот файлы изображений http://www.mediafire.com/file/emm0vhhuwpwdx6v/exr_files.zip/file

1 Ответ

0 голосов
/ 19 октября 2018

Поскольку я не могу воспроизвести проблему, поскольку вы не предоставляете MCVE, я не смогу указать на ошибку, поэтому я покажу вам рабочий код, в этом случае я буду считать, что изображения включены.сторона .py: 

├── chip.exr
├── main.py
└── panel.exr

код: 

import OpenEXR, Imath
from PIL import Image
import array
from PIL.ImageQt import ImageQt as PilImageQt
from PyQt5 import QtCore, QtGui, QtWidgets


def exrToJpgGamma(exrfile):
    file = OpenEXR.InputFile(exrfile)
    pt = Imath.PixelType(Imath.PixelType.FLOAT)
    dw = file.header()['dataWindow']
    size = (dw.max.x - dw.min.x + 1, dw.max.y - dw.min.y + 1)
    RedStr = file.channel('R', pt)
    GreenStr = file.channel('G', pt)
    BlueStr = file.channel('B', pt)
    Red = array.array('f', RedStr)
    Green = array.array('f', GreenStr)
    Blue = array.array('f', BlueStr)
    def EncodeToSRGB(v):
        if v <= 0.0031308:
            return (v * 12.92) * 255.0
        else:
            return (1.055*(v**(1.0/2.2))-0.055) * 255.0
    for I, value in enumerate(Red):
        Red[I] = EncodeToSRGB(value)
    for I, value in enumerate(Green):
        Green[I] = EncodeToSRGB(value)
    for I, value in enumerate(Blue):
        Blue[I] = EncodeToSRGB(value)
    rgbf = [Image.frombytes("F", size, channel.tobytes()) for channel in (Red, Green, Blue)]
    rgb8 = [im.convert("L") for im in rgbf]
    myqimage = Image.merge("RGB", rgb8)
    return myqimage

class Widget(QtWidgets.QWidget):
    def __init__(self, parent=None):
        super(Widget, self).__init__(parent)
        self.chip_img = QtWidgets.QLabel()
        self.panel_img = QtWidgets.QLabel()
        lay = QtWidgets.QVBoxLayout(self)
        lay.addWidget(self.chip_img)
        lay.addWidget(self.panel_img)

        self.load_exr('panel.exr', self.panel_img)
        self.load_exr('chip.exr', self.chip_img)

    def load_exr(self, filename, label):
        width, height = 480, 360
        imageq = PilImageQt(exrToJpgGamma(filename))
        pixmap = QtGui.QPixmap.fromImage(imageq.copy())
        scaled_pixmap = pixmap.scaled(width, height, 
            QtCore.Qt.KeepAspectRatio, 
            QtCore.Qt.FastTransformation)
        label.setPixmap(scaled_pixmap)

if __name__ == '__main__':
    import sys

    app = QtWidgets.QApplication(sys.argv)
    w = Widget()
    w.show()
    sys.exit(app.exec_())

enter image description here

...