利用Python将EXR格式文件转换为HDR文件

主要是利用OpenEXR库读取EXR文件，然后利用imageio库来写回HDR

import numpy
import OpenEXR
import Imath
import imageio
import glob
import os


def ext2hdr(exrpath):
    File = OpenEXR.InputFile(exrpath)
    PixType = Imath.PixelType(Imath.PixelType.FLOAT)
    DW = File.header()['dataWindow']
    Size = (DW.max.x - DW.min.x + 1, DW.max.y - DW.min.y + 1)
    rgb = [numpy.fromstring(File.channel(c, PixType), dtype=numpy.float32) for c in 'RGB']
    r =numpy.reshape(rgb[0],(Size[1],Size[0]))
    g =numpy.reshape(rgb[1],(Size[1],Size[0]))
    b =numpy.reshape(rgb[2],(Size[1],Size[0]))
    hdr = numpy.zeros((Size[1],Size[0],3),dtype=numpy.float32)
    hdr[:,:,0] = r
    hdr[:,:,1] = g
    hdr[:,:,2] = b
    return hdr

def writehdr(hdrpath,hdr):
    imageio.imwrite(hdrpath,hdr,format='hdr')




files = glob.glob('/home/jie/PycharmProjects/CNNTM/data/Fairchild/*.exr')
savepath = '/home/jie/PycharmProjects/CNNTM/data/Fairchild/HDR_format'
for file in files:
    hdr = ext2hdr(file)
    filename,file_ext = os.path.splitext(file)
    filename = os.path.basename(filename)
    filename = filename + '.hdr'
    curpath = os.path.join(savepath,filename)
    writehdr(curpath,hdr)