Myosotis-Researches 0.1.31__tar.gz → 0.1.33__tar.gz

{myosotis_researches-0.1.31 → myosotis_researches-0.1.33/Myosotis_Researches.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.1.31
+Version: 0.1.33
 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.31 → myosotis_researches-0.1.33}/Myosotis_Researches.egg-info/SOURCES.txt

@@ -38,6 +38,7 @@ myosotis_researches/CcGAN/train/train_net_for_label_embed.py
 myosotis_researches/CcGAN/utils/__init__.py
 myosotis_researches/CcGAN/utils/concat_image.py
 myosotis_researches/CcGAN/utils/dataset.py
+myosotis_researches/CcGAN/utils/similarity.py
 myosotis_researches/CcGAN/utils/src/style.css
 myosotis_researches/CcGAN/utils/src/style.css.map
 myosotis_researches/CcGAN/utils/src/style.scss

{myosotis_researches-0.1.31/Myosotis_Researches.egg-info → myosotis_researches-0.1.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.1.31
+Version: 0.1.33
 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.31 → myosotis_researches-0.1.33}/myosotis_researches/CcGAN/utils/__init__.py

@@ -1,2 +1,3 @@
 from .concat_image import concat_image
 from .dataset import view_dataset, visualize_dataset, download_datasets
+from .similarity import similarity

myosotis_researches-0.1.33/myosotis_researches/CcGAN/utils/similarity.py

@@ -0,0 +1,52 @@
+import cv2
+import numpy as np
+from PIL import Image
+from skimage.metrics import structural_similarity as ssim
+
+
+# MSE
+def _MSE(image_1, image_2):
+    array_1 = np.array(image_1) / 255
+    array_2 = np.array(image_2) / 255
+    n, m, _ = array_1.shape
+    return np.sum((array_1 - array_2) ** 2) / (n * m * 3)
+
+
+def MSE(images):
+    MSE_sum = 0
+    n = len(images)
+    for i in range(n):
+        for j in range(i + 1, n):
+            MSE_sum += _MSE(images[i], images[j])
+    return MSE_sum / (n * (n - 1) / 2)
+
+
+# SSIM
+
+
+def _SSIM(image_1, image_2):
+
+    array_1 = np.array(image_1)
+    array_2 = np.array(image_2)
+    score, _ = ssim(array_1, array_2, channel_axis=-1, full=True)
+    return score
+
+
+def SSIM(images):
+    SSIM_sum = 0
+    n = len(images)
+    for i in range(n):
+        for j in range(i + 1, n):
+            SSIM_sum += _SSIM(images[i], images[j])
+    return SSIM_sum / (n * (n - 1) / 2)
+
+
+# Similarity
+_method_dict = {"MSE": MSE, "SSIM": SSIM}
+
+
+def similarity(images, method):
+    return _method_dict[method](images)
+
+
+__all__ = ["similarity"]

{myosotis_researches-0.1.31 → myosotis_researches-0.1.33}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="Myosotis-Researches",
-    version="0.1.31",
+    version="0.1.33",
     description="A repository for storing my progress of researches.",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",