36 lines
946 B
Python
36 lines
946 B
Python
|
import cv2
|
||
|
|
import numpy as np
|
||
|
|
import skimage
|
||
|
|
|
||
|
|
def gamma_correction(image, gamma):
|
||
|
|
adjust = skimage.exposure.rescale_intensity(image)
|
||
|
|
adjust = skimage.exposure.adjust_gamma(adjust,gamma)
|
||
|
|
return adjust
|
||
|
|
|
||
|
|
def fusionhdr(images: list):
|
||
|
|
# Load the set of images
|
||
|
|
|
||
|
|
img_array = []
|
||
|
|
for img in images:
|
||
|
|
img_array.append(cv2.imread(img))
|
||
|
|
|
||
|
|
# Create Mertens object and set its parameters
|
||
|
|
mertens = cv2.createMergeMertens()
|
||
|
|
mertens.setExposureWeight(0.9)
|
||
|
|
mertens.setSaturationWeight(0.5)
|
||
|
|
|
||
|
|
# Merge the images using the Mertens algorithm
|
||
|
|
fused_image = mertens.process(img_array)
|
||
|
|
fused = np.clip(fused_image, 0, 1)
|
||
|
|
|
||
|
|
# Save the fused image
|
||
|
|
|
||
|
|
return fused
|
||
|
|
|
||
|
|
list = ['hdr/webcam20_3_2023_ev1.jpg', 'hdr/webcam20_3_2023_ev0.jpg','hdr/webcam20_3_2023_ev-1.jpg']
|
||
|
|
|
||
|
|
#TODO add image open
|
||
|
|
hdr_image = fusionhdr(list)
|
||
|
|
hdr_image = gamma_correction(hdr_image,0.7)
|
||
|
|
cv2.imwrite('fused_image2.jpg', hdr_image * 255)
|