Чтение.exr файлов в OpenCV

Question

Я сгенерировал несколько карт глубины, используя blender, и сохранил значения z-буфера (32 бита) в формате OpenEXR. Есть ли способ получить доступ к значениям из файла.exr (информация о глубине пикселя), используя OpenCV 2.4.13 и python 2.7? Там нет нигде примера, чтобы найти. Все, что я вижу в документации, что этот формат файла поддерживается. Но попытка прочитать такой файл приводит к ошибке.

new=cv2.imread("D:\\Test1\\0001.exr")
cv2.imshow('exr',new)
print new[0,0]

Ошибка:

print new[0,0]
TypeError: 'NoneType' object has no attribute '__getitem__'

а также

cv2.imshow('exr',new)
cv2.error: ..\..\..\..\opencv\modules\highgui\src\window.cpp:261: error: (-215) size.width>0 && size.height>0 in function cv::imshow

Ближе всего я нашел эту ссылку и эту ссылку.

Answer 1

Я мог бы немного опоздать на вечеринку, но; Да, вы определенно можете использовать OpenCV для этого.

cv2.imread(PATH_TO_EXR_FILE,  cv2.IMREAD_ANYCOLOR | cv2.IMREAD_ANYDEPTH)  

должен получить то, что тебе нужно

Answer 2

Полное решение

Решение @Iwohlhart вызвало у меня ошибку, а затем исправило ее,

      # first import os and enable the necessary flags to avoid cv2 errors

import os
os.environ["OPENCV_IO_ENABLE_OPENEXR"]="1"
import cv2

# then just type in following

img = cv2.imread(PATH2EXR, cv2.IMREAD_ANYCOLOR | cv2.IMREAD_ANYDEPTH)
'''
you might have to disable following flags, if you are reading a semantic map/label then because it will convert it into binary map so check both lines and see what you need
''' 
# img = cv2.imread(PATH2EXR) 
 
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

Answer 3

Вы можете использовать пакет OpenEXR

pip --no-cache-dir install OpenEXR

Чтобы прочитать exr файл

def read_depth_exr_file(filepath: Path):
    exrfile = exr.InputFile(filepath.as_posix())
    raw_bytes = exrfile.channel('B', Imath.PixelType(Imath.PixelType.FLOAT))
    depth_vector = numpy.frombuffer(raw_bytes, dtype=numpy.float32)
    height = exrfile.header()['displayWindow'].max.y + 1 - exrfile.header()['displayWindow'].min.y
    width = exrfile.header()['displayWindow'].max.x + 1 - exrfile.header()['displayWindow'].min.x
    depth_map = numpy.reshape(depth_vector, (height, width))
    return depth_map