Myosotis-Researches 0.1.40__py3-none-any.whl → 0.1.42__py3-none-any.whl

myosotis_researches/CcGAN/utils/similarity.py

@@ -2,6 +2,7 @@ import cv2
 import math
 import numpy as np
 from PIL import Image
+from scipy.special import rel_entr
 from skimage.metrics import structural_similarity as ssim
 
 
@@ -92,12 +93,14 @@ def cross_correlation(images):
 def _rgb_histogram(image, bins=32):
     """
     image: numpy array of shape (H, W, 3), RGB format
-    returns: concatenated histogram for R, G, B channels
+    returns: concatenated and normalized histogram for R, G, B channels
     """
-    hist_r, _ = np.histogram(image[:, :, 0], bins=bins, range=(0, 255))
-    hist_g, _ = np.histogram(image[:, :, 1], bins=bins, range=(0, 255))
-    hist_b, _ = np.histogram(image[:, :, 2], bins=bins, range=(0, 255))
-    hist = np.concatenate([hist_r, hist_g, hist_b])
+    image = np.asarray(image)
+    hist_r, _ = np.histogram(image[:, :, 0], bins=bins, range=(0, 256))
+    hist_g, _ = np.histogram(image[:, :, 1], bins=bins, range=(0, 256))
+    hist_b, _ = np.histogram(image[:, :, 2], bins=bins, range=(0, 256))
+    hist = np.concatenate([hist_r, hist_g, hist_b]).astype(np.float64)
+    hist /= np.sum(hist)  # Normalize to make it a probability distribution
     return hist
 
 
@@ -118,6 +121,33 @@ def histogram_intersection(images):
     return histogram_intersection_sum / (n * (n - 1) / 2)
 
 
+# KL divergence
+def _kl_divergence(image_1, image_2, bins=256):
+    kl_values = []
+
+    for c in range(3):
+        arr1 = np.array(image_1)[:, :, c].flatten()
+        arr2 = np.array(image_2)[:, :, c].flatten()
+        hist1, _ = np.histogram(arr1, bins=bins, range=(0, 256), density=True)
+        hist2, _ = np.histogram(arr2, bins=bins, range=(0, 256), density=True)
+        hist1 += 1e-10
+        hist2 += 1e-10
+        kl_1_2 = np.sum(rel_entr(hist1, hist2))
+        kl_2_1 = np.sum(rel_entr(hist2, hist1))
+        sym_kl = 0.5 * (kl_1_2 + kl_2_1)
+        kl_values.append(sym_kl)
+    return float(np.mean(kl_values))
+
+
+def kl_divergence(images):
+    kl_divergence_sum = 0
+    n = len(images)
+    for i in range(n):
+        for j in range(i + 1, n):
+            kl_divergence_sum += _kl_divergence(images[i], images[j])
+    return kl_divergence_sum / (n * (n - 1) / 2)
+
+
 # Similarity
 _method_dict = {
     "MSE": MSE,
@@ -125,6 +155,7 @@ _method_dict = {
     "PSNR": PSNR,
     "cross-correlation": cross_correlation,
     "histogram-intersection": histogram_intersection,
+    "kl-divergence": kl_divergence,
 }
 
 

{myosotis_researches-0.1.40.dist-info → myosotis_researches-0.1.42.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.1.40
+Version: 0.1.42
 Summary: A repository for storing my progress of researches.
 Home-page: https://github.com/Zeyu-Xie/Myosotis-Researches
 Author: Zeyu Xie

{myosotis_researches-0.1.40.dist-info → myosotis_researches-0.1.42.dist-info}/RECORD

@@ -29,13 +29,13 @@ myosotis_researches/CcGAN/train/train_net_for_label_embed.py,sha256=4j6r4_o4rXgA
 myosotis_researches/CcGAN/utils/__init__.py,sha256=yH4I2QzSoBsjcdjvOH0YKLTeEji_oWrzBGm5EkVZpoI,146
 myosotis_researches/CcGAN/utils/concat_image.py,sha256=BIGKz52Inn9S7M5fBFKye2V9bLJ0DqEQILoOVWAXUiE,2165
 myosotis_researches/CcGAN/utils/dataset.py,sha256=TXJSLMpwI1h0xb6lXpq5iUjjurheKPZnPbR2KIU-Py4,4216
-myosotis_researches/CcGAN/utils/similarity.py,sha256=q9Z1G7nxDViAGQWRVy3hAncPOVTdC2aEKq3UByQcJAE,3496
+myosotis_researches/CcGAN/utils/similarity.py,sha256=9YMcHgebmPXn7R4fhFw9tBG1CcYxv102F9G_nCI5j4k,4582
 myosotis_researches/CcGAN/utils/src/style.css,sha256=bNp3_nlpg0W5qA7Rx8MM5ayeprEbYH3_6AIDRTN8UhM,566
 myosotis_researches/CcGAN/utils/src/style.css.map,sha256=GTGPet_xXnKQnsz-E2CMUWtrgu3D5t0g4J_A_amFdvw,246
 myosotis_researches/CcGAN/utils/src/style.scss,sha256=_Wa7VPmFzCkEHtrzdG35tj89QNwPd-fpiT-YL4ohcMs,526
 myosotis_researches/CcGAN/utils/src/template.html,sha256=S35ak_uJ0Zd3_PhX-YoCRBP2TMKxdFPMMaBdpARdKcU,3890
-myosotis_researches-0.1.40.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-myosotis_researches-0.1.40.dist-info/METADATA,sha256=e6btj8xrHJ7RFkvM8Ar148cCj7etp0_EmAXG79yA4qk,3484
-myosotis_researches-0.1.40.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-myosotis_researches-0.1.40.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
-myosotis_researches-0.1.40.dist-info/RECORD,,
+myosotis_researches-0.1.42.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+myosotis_researches-0.1.42.dist-info/METADATA,sha256=ZOvU_PxDnsWcScz8Xtbb7rISgRJhHDNFvB28JZtQ9Hg,3484
+myosotis_researches-0.1.42.dist-info/WHEEL,sha256=GHB6lJx2juba1wDgXDNlMTyM13ckjBMKf-OnwgKOCtA,91
+myosotis_researches-0.1.42.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
+myosotis_researches-0.1.42.dist-info/RECORD,,

{myosotis_researches-0.1.40.dist-info → myosotis_researches-0.1.42.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.1.0)
+Generator: setuptools (80.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any