Added lots of stuff

This commit is contained in:
lefreud committed 2024-01-17 19:03:04 +00:00
1 parent 3aadde70eb
commit e602d69c43
14 files changed
+83 -33

No files matched your search

+4 -1
View File
@@ -8,7 +8,10 @@ import numpy as np
# fonction that apply the correction on the colors on the input images
def apply_correction(coefs, img, CORRECTION_CURVE_TYPE) -> np.array:
def apply_correction(coefs, img, CORRECTION_CURVE_TYPE) -> np.ndarray:
# check that img is in range 0-255
assert img.dtype == np.uint8
B, G, R = coefs
original_shape = img.shape
img = img.astype(np.float32)