color grading

This commit is contained in:
Vladimir Mandic
2024-02-05 13:40:00 -05:00
parent b8ea66b262
commit 3b4d59877f
6 changed files with 20 additions and 16 deletions
+2 -2
View File
@@ -75,9 +75,9 @@ def get_color(colorstr):
def color_adjust(tensor, colorstr, ratio):
color = get_color(colorstr)
debug(f'HDR tint: str={colorstr} color={color} ratio={ratio}')
for i in range(3):
print(color[i])
tensor[i] = center_tensor(tensor[i], full_shift=1, offset=color[i]*(ratio/3))
tensor[i] = center_tensor(tensor[i], full_shift=1, offset=color[i]*(ratio/2))
return tensor