Spaces:
Runtime error
Runtime error
File size: 2,424 Bytes
421360e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import numpy as np
import cv2
from PIL import Image
# def read_img(image_path):
# img = cv2.imread(image_path)
# img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
# assert len(img.shape)==3
# return img
def read_img(image_file_buffer):
img = Image.open(image_file_buffer).convert('RGB')
img = np.array(img)
assert len(img.shape)==3
return img
# Calculates the average brightness in the specified irregular image
def calculate_average_brightness(img):
# Average value of three color channels
R = img[..., 0].mean()
G = img[..., 1].mean()
B = img[..., 2].mean()
brightness = 0.299 * R + 0.587 * G + 0.114 * B
return brightness, B, G, R
# Adjusting the average brightness of the target image to the average brightness of the source image
def adjust_brightness_from_src_to_dst(dst, src,path=None,if_show=None, if_info=None):
brightness1, B1, G1, R1 = calculate_average_brightness(src)
brightness2, B2, G2, R2 = calculate_average_brightness(dst)
brightness_difference = brightness1 / brightness2
if if_info:
print('Average brightness of original image', brightness1)
print('Average brightness of target', brightness2)
print('Brightness Difference between Original Image and Target', brightness_difference)
# According to the average display brightness
dstf = dst * brightness_difference
# According to the average value of the three-color channel
# dstf = dst.copy().astype(np.float32)
# dstf[..., 0] = dst[..., 0] * (B1 / B2)
# dstf[..., 1] = dst[..., 1] * (G1 / G2)
# dstf[..., 2] = dst[..., 2] * (R1 / R2)
# To limit the results and prevent crossing the border,
# it must be converted to uint8, otherwise the default result is float32, and errors will occur.
dstf = np.clip(dstf, 0, 255)
dstf = np.uint8(dstf)
ma,na,_ = src.shape
mb,nb,_ = dst.shape
result_show_img = np.zeros((max(ma, mb), 3 * max(na, nb), 3))
result_show_img[:mb, :nb, :] = dst
result_show_img[:ma, nb:nb + na, :] = src
result_show_img[:mb, nb + na:nb + na + nb, :] = dstf
result_show_img = result_show_img.astype(np.uint8)
if if_show:
cv2.imshow('-', cv2.cvtColor(result_show_img, cv2.COLOR_BGR2RGB))
cv2.waitKey(0)
cv2.destroyAllWindows()
if path != None:
cv2.imwrite(path, cv2.cvtColor(result_show_img, cv2.COLOR_BGR2RGB))
return dstf
|