主要是利用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)
