From c2f2b183aa7a80df078dd9914c66a944d75acf60 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Sun, 16 Jun 2024 20:50:59 +0100 Subject: [PATCH] Remove clutter --- src/editor/utils/__init__.py | 6 ++++-- src/editor/utils/kldiv.py | 11 ----------- 2 files changed, 4 insertions(+), 13 deletions(-) delete mode 100644 src/editor/utils/kldiv.py diff --git a/src/editor/utils/__init__.py b/src/editor/utils/__init__.py index 845297c..5f8fc9a 100644 --- a/src/editor/utils/__init__.py +++ b/src/editor/utils/__init__.py @@ -1,8 +1,10 @@ from .interpolate import interpolate from .random import random from .apply_pixel_shader import apply_pixel_shader -from .get_colour_lut import get_colour_lut +from .get_colour_lut import ( + get_random_brightness_lut, + get_random_saturation_per_hue_lut, +) from .compute_histogram import compute_histogram -from .kldiv import kldiv from .generate_rotation_matrices import generate_rotation_matrices from .get_next_run_name import get_next_run_name diff --git a/src/editor/utils/kldiv.py b/src/editor/utils/kldiv.py deleted file mode 100644 index 1f85d25..0000000 --- a/src/editor/utils/kldiv.py +++ /dev/null @@ -1,11 +0,0 @@ -import numpy as np - - -def kldiv(P: np.ndarray, Q: np.ndarray) -> float: - P /= P.sum() - Q /= Q.sum() - - P_safe = np.maximum(P, np.finfo(float).eps) - Q_safe = np.maximum(Q, np.finfo(float).eps) - - return np.sum(P_safe * np.log(P_safe / Q_safe))