Конвертировать EXR в JPEG с использованием ImageIO и Python

Question

Я просто пытаюсь конвертировать и EXR в изображение JPG, однако мои результаты оказываются очень темными. Кто-нибудь знает, что я здесь делаю не так? Я нормализую значения изображения и затем помещаю их в цветовое пространство 0-255. Это все еще кажется неправильным, хотя.

Ссылка Dropbox для проверки изображения exr: https://www.dropbox.com/s/9a5z6fjsyth7w98/torus.exr?dl=0

import sys, os
import imageio

def convert_exr_to_jpg(exr_file, jpg_file):
    if not os.path.isfile(exr_file):
        return False

    filename, extension = os.path.splitext(exr_file)
    if not extension.lower().endswith('.exr'):
        return False

    # imageio.plugins.freeimage.download() #DOWNLOAD IT
    image = imageio.imread(exr_file, format='EXR-FI')

    # remove alpha channel for jpg conversion
    image = image[:,:,:3]

    # normalize the image
    data = image.astype(image.dtype) / image.max() # normalize the data to 0 - 1
    data = 255 * data # Now scale by 255
    rgb_image = data.astype('uint8')
    # rgb_image = imageio.core.image_as_uint(rgb_image, bitdepth=8)

    imageio.imwrite(jpg_file, rgb_image, format='jpeg')
    return True


if __name__ == '__main__':
    exr = "C:/Users/John/images/torus.exr"
    jpg = "C:/Users/John/images/torus.jpg"
    convert_exr_to_jpg(exr, jpg)

Answer 1

Образец изображения представляет собой изображение EXR с глубиной (каналом) 16 бит. Вот сценарий python для преобразования изображения exr в png с помощьюopencv.

import numpy as np
import cv2
im=cv2.imread("torus.exr",-1)
im=im*65535
im[im>65535]=65535
im=np.uint16(im)
cv2.imwrite("torus.png",im)

Вот модифицированный код сimageio, который сохранит изображение в форматеjpeg.

import sys, os
import imageio

def convert_exr_to_jpg(exr_file, jpg_file):
    if not os.path.isfile(exr_file):
        return False

    filename, extension = os.path.splitext(exr_file)
    if not extension.lower().endswith('.exr'):
        return False

    # imageio.plugins.freeimage.download() #DOWNLOAD IT
    image = imageio.imread(exr_file)
    print(image.dtype)

    # remove alpha channel for jpg conversion
    image = image[:,:,:3]


    data = 65535 * image
    data[data>65535]=65535
    rgb_image = data.astype('uint16')
    print(rgb_image.dtype)
    #rgb_image = imageio.core.image_as_uint(rgb_image, bitdepth=16)

    imageio.imwrite(jpg_file, rgb_image, format='jpeg')
    return True


if __name__ == '__main__':
    exr = "torus.exr"
    jpg = "torus3.jpeg"
    convert_exr_to_jpg(exr, jpg)

(Протестировано с python 3.5.2,Ubuntu 16.04)

Answer 2

Я столкнулся с той же проблемой и исправил ее здесь. Поскольку ImageIO преобразует все в массив numpy, вы можете гамма-корректировать значения (устраняя проблему с темнотой), а затем преобразовать это обратно в изображение PIL для легкой работы:

im = imageio.imread("image.exr")
im_gamma_correct = numpy.clip(numpy.power(im, 0.45), 0, 1)
im_fixed = Image.fromarray(numpy.uint8(im_gamma_correct*255))

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