Fast-SSIM 1.0.0__tar.gz

fast_ssim-1.0.0/Fast_SSIM.egg-info/PKG-INFO

@@ -0,0 +1,84 @@
+Metadata-Version: 2.4
+Name: Fast-SSIM
+Version: 1.0.0
+Summary: Fast SSIM and PSNR algorithms for Python. Up to 30x speedup for SSIM and up to 10x for PSNR.
+Author: Tim Lodemann
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/timminator/Fast-SSIM
+Project-URL: documentation, https://github.com/timminator/Fast-SSIM/blob/master/README.md
+Project-URL: Repository, https://github.com/timminator/Fast-SSIM.git
+Project-URL: Issues, https://github.com/timminator/Fast-SSIM/issues
+Keywords: ssim,psnr,image quality,image comparison,image processing
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Scientific/Engineering :: Image Processing
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Dynamic: license-file
+
+<p align="center">
+  <h1 align="center">Fast-SSIM</h1>
+  <p align="center">
+    Speed up your SSIM and PSNR calculations!
+    <br />
+  </p>
+</p>
+
+<br>
+
+## ℹ About
+
+This fork wraps the Fast-SSIM project in an easy to use package that can be readily installed via PyPI.
+The Fast-SSIM package can accelerate your SSIM and PSNR calculations by up to 30x and 10x respectively.
+
+## Requirements
+
+- Python 3.9 or higher
+
+## How to Install
+
+```
+pip install Fast-SSIM
+```
+
+## Usage
+
+The functionalities are explained in the following code snippet that is also provided in this repo:
+
+```python
+import fast_ssim
+from skimage.io import imread
+
+img1_path = r"test_images\0.jpg"
+img2_path = r"test_images\1.jpg"
+
+# Load the images into NumPy arrays
+img1 = imread(img1_path)
+img2 = imread(img2_path)
+
+ssim_score = fast_ssim.ssim(
+    img1, img2,
+    data_range=255
+)
+psnr_score = fast_ssim.psnr(
+    img1, img2,
+    data_range=255
+)
+
+print(f"SSIM Score: {ssim_score}")
+print(f"PSNR Score: {psnr_score}")
+```
+
+Output:
+```
+SSIM Score: 0.9886590838432312
+PSNR Score: 31.11587142944336
+```
+
+## Original author
+
+Chen Yu / [@Chen Yu](https://github.com/chinue)

fast_ssim-1.0.0/Fast_SSIM.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+LICENSE
+README.md
+pyproject.toml
+Fast_SSIM.egg-info/PKG-INFO
+Fast_SSIM.egg-info/SOURCES.txt
+Fast_SSIM.egg-info/dependency_links.txt
+Fast_SSIM.egg-info/requires.txt
+Fast_SSIM.egg-info/top_level.txt
+fast_ssim/__init__.py
+fast_ssim/_core.py
+fast_ssim/resources/libssim.so
+fast_ssim/resources/ssim.dll

fast_ssim-1.0.0/Fast_SSIM.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

fast_ssim-1.0.0/Fast_SSIM.egg-info/requires.txt

@@ -0,0 +1 @@
+numpy

fast_ssim-1.0.0/Fast_SSIM.egg-info/top_level.txt

@@ -0,0 +1 @@
+fast_ssim

fast_ssim-1.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2022 chinue
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

fast_ssim-1.0.0/PKG-INFO

@@ -0,0 +1,84 @@
+Metadata-Version: 2.4
+Name: Fast-SSIM
+Version: 1.0.0
+Summary: Fast SSIM and PSNR algorithms for Python. Up to 30x speedup for SSIM and up to 10x for PSNR.
+Author: Tim Lodemann
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/timminator/Fast-SSIM
+Project-URL: documentation, https://github.com/timminator/Fast-SSIM/blob/master/README.md
+Project-URL: Repository, https://github.com/timminator/Fast-SSIM.git
+Project-URL: Issues, https://github.com/timminator/Fast-SSIM/issues
+Keywords: ssim,psnr,image quality,image comparison,image processing
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Scientific/Engineering :: Image Processing
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Dynamic: license-file
+
+<p align="center">
+  <h1 align="center">Fast-SSIM</h1>
+  <p align="center">
+    Speed up your SSIM and PSNR calculations!
+    <br />
+  </p>
+</p>
+
+<br>
+
+## ℹ About
+
+This fork wraps the Fast-SSIM project in an easy to use package that can be readily installed via PyPI.
+The Fast-SSIM package can accelerate your SSIM and PSNR calculations by up to 30x and 10x respectively.
+
+## Requirements
+
+- Python 3.9 or higher
+
+## How to Install
+
+```
+pip install Fast-SSIM
+```
+
+## Usage
+
+The functionalities are explained in the following code snippet that is also provided in this repo:
+
+```python
+import fast_ssim
+from skimage.io import imread
+
+img1_path = r"test_images\0.jpg"
+img2_path = r"test_images\1.jpg"
+
+# Load the images into NumPy arrays
+img1 = imread(img1_path)
+img2 = imread(img2_path)
+
+ssim_score = fast_ssim.ssim(
+    img1, img2,
+    data_range=255
+)
+psnr_score = fast_ssim.psnr(
+    img1, img2,
+    data_range=255
+)
+
+print(f"SSIM Score: {ssim_score}")
+print(f"PSNR Score: {psnr_score}")
+```
+
+Output:
+```
+SSIM Score: 0.9886590838432312
+PSNR Score: 31.11587142944336
+```
+
+## Original author
+
+Chen Yu / [@Chen Yu](https://github.com/chinue)

fast_ssim-1.0.0/README.md

@@ -0,0 +1,62 @@
+<p align="center">
+  <h1 align="center">Fast-SSIM</h1>
+  <p align="center">
+    Speed up your SSIM and PSNR calculations!
+    <br />
+  </p>
+</p>
+
+<br>
+
+## ℹ About
+
+This fork wraps the Fast-SSIM project in an easy to use package that can be readily installed via PyPI.
+The Fast-SSIM package can accelerate your SSIM and PSNR calculations by up to 30x and 10x respectively.
+
+## Requirements
+
+- Python 3.9 or higher
+
+## How to Install
+
+```
+pip install Fast-SSIM
+```
+
+## Usage
+
+The functionalities are explained in the following code snippet that is also provided in this repo:
+
+```python
+import fast_ssim
+from skimage.io import imread
+
+img1_path = r"test_images\0.jpg"
+img2_path = r"test_images\1.jpg"
+
+# Load the images into NumPy arrays
+img1 = imread(img1_path)
+img2 = imread(img2_path)
+
+ssim_score = fast_ssim.ssim(
+    img1, img2,
+    data_range=255
+)
+psnr_score = fast_ssim.psnr(
+    img1, img2,
+    data_range=255
+)
+
+print(f"SSIM Score: {ssim_score}")
+print(f"PSNR Score: {psnr_score}")
+```
+
+Output:
+```
+SSIM Score: 0.9886590838432312
+PSNR Score: 31.11587142944336
+```
+
+## Original author
+
+Chen Yu / [@Chen Yu](https://github.com/chinue)

fast_ssim-1.0.0/fast_ssim/__init__.py

@@ -0,0 +1 @@
+from ._core import ssim, psnr

fast_ssim-1.0.0/fast_ssim/_core.py

@@ -0,0 +1,98 @@
+import numpy as np
+import ctypes
+import os
+
+__version__ = '1.0.0'
+
+ssim_dll_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'resources')
+ssim_dll_name = 'ssim.dll' if(os.name=='nt') else 'libssim.so'
+
+
+class SharedLibraryLoadError(ImportError):
+    """Raised when the shared SSIM/PSNR library could not be loaded."""
+    pass
+
+
+class Loader:
+    dll = None
+
+    try:
+        dll_path = os.path.join(ssim_dll_path, ssim_dll_name)
+        if os.path.exists(dll_path):
+            dll = np.ctypeslib.load_library(ssim_dll_name, ssim_dll_path)
+        else:
+            raise SharedLibraryLoadError(f"Shared library not found at: {dll_path}")
+    except Exception as e:
+        raise SharedLibraryLoadError(f"Failed to load shared library '{ssim_dll_name}': {e}") from e
+
+    type_dict = {'int':ctypes.c_int, 'float':ctypes.c_float, 'double':ctypes.c_double, 'void':None,
+                    'int32':ctypes.c_int32, 'uint32':ctypes.c_uint32, 'int16':ctypes.c_int16, 'uint16':ctypes.c_uint16,
+                    'int8':ctypes.c_int8, 'uint8':ctypes.c_uint8, 'byte':ctypes.c_uint8,
+                    'char*':ctypes.c_char_p,
+                    'float*':np.ctypeslib.ndpointer(dtype='float32', ndim=1, flags='CONTIGUOUS'),
+                    'int*':np.ctypeslib.ndpointer(dtype='int32', ndim=1, flags='CONTIGUOUS'),
+                    'byte*':np.ctypeslib.ndpointer(dtype='uint8', ndim=1, flags='CONTIGUOUS')}
+
+    @staticmethod
+    def get_function(res_type='float', func_name='PSNR_Byte', arg_types=['Byte*', 'int', 'int', 'int', 'Byte*']):
+        func = Loader.dll.__getattr__(func_name)
+        func.restype = Loader.type_dict[res_type]
+        func.argtypes = [Loader.type_dict[str.lower(x).replace(' ', '')] for x in arg_types]
+        return func
+
+    @staticmethod
+    def had_member(name='dll'):
+        return (name in Loader.__dict__.keys())
+
+
+class DLL:
+    @staticmethod
+    def had_function(name='PSNR_Byte'):
+        return (name in DLL.__dict__.keys())
+
+    if(Loader.had_member('dll')):
+        # float PSNR_Byte(Byte* pDataX, Byte* pDataY, int step, int width, int height, int maxVal);
+        PSNR_Byte = Loader.get_function('float', 'PSNR_Byte', ['Byte*', 'Byte*', 'int', 'int', 'int', 'int'])
+
+        # float PSNR_Float(float* pDataX, float* pDataY, int step, int width, int height, double maxVal);
+        PSNR_Float = Loader.get_function('float', 'PSNR_Float', ['float*', 'float*', 'int', 'int', 'int', 'double'])
+
+        # float SSIM_Byte(Byte* pDataX, Byte* pDataY, int step, int width, int height, int win_size, int maxVal);
+        SSIM_Byte = Loader.get_function('float', 'SSIM_Byte', ['Byte*', 'Byte*', 'int', 'int', 'int', 'int', 'int'])
+
+        # float SSIM_Float(float* pDataX, float* pDataY, int step, int width, int height, int win_size, double maxVal);
+        SSIM_Float = Loader.get_function('float', 'SSIM_Float', ['float*', 'float*', 'int', 'int', 'int', 'int', 'double'])
+
+
+def psnr(x, y, data_range=None):
+    if x.ndim == 2:
+        h, w = x.shape
+        c = 1
+    elif x.ndim == 3:
+        h, w, c = x.shape
+    else:
+        raise ValueError(f"Unsupported image dimensions: {x.shape}")
+
+    x = x.astype('float32') if(x.dtype=='float64') else x
+    y = y.astype('float32') if(y.dtype=='float64') else y
+    if(DLL.had_function('PSNR_Byte') and x.dtype=='uint8' and y.dtype=='uint8'):
+        return DLL.PSNR_Byte(x.reshape([-1]), y.reshape([-1]), w*c, w, h, 255 if(data_range==None) else int(data_range))
+    if(DLL.had_function('PSNR_Float') and x.dtype=='float32' and y.dtype=='float32'):
+        return DLL.PSNR_Float(x.reshape([-1]), y.reshape([-1]), w*c, w, h, 255.0 if(data_range==None) else float(data_range))
+
+
+def ssim(x, y, data_range=None, win_size=7):
+    if x.ndim == 2:
+        h, w = x.shape
+        c = 1
+    elif x.ndim == 3:
+        h, w, c = x.shape
+    else:
+        raise ValueError(f"Unsupported image dimensions: {x.shape}")
+
+    x = x.astype('float32') if(x.dtype=='float64') else x
+    y = y.astype('float32') if(y.dtype=='float64') else y
+    if(DLL.had_function('SSIM_Byte') and x.dtype=='uint8' and y.dtype=='uint8'):
+        return DLL.SSIM_Byte(x.reshape([-1]), y.reshape([-1]), w*c, w, h, win_size, 255 if(data_range==None) else int(data_range))
+    if(DLL.had_function('SSIM_Float') and x.dtype=='float32' and y.dtype=='float32'):
+        return DLL.SSIM_Float(x.reshape([-1]), y.reshape([-1]), w*c, w, h, win_size, 255.0 if(data_range==None) else float(data_range))

fast_ssim-1.0.0/pyproject.toml

@@ -0,0 +1,37 @@
+[build-system]
+requires = ["setuptools>=77.0.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "Fast-SSIM"
+description = "Fast SSIM and PSNR algorithms for Python. Up to 30x speedup for SSIM and up to 10x for PSNR."
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+keywords = ["ssim", "psnr", "image quality", "image comparison", "image processing"]
+authors = [{name = "Tim Lodemann"}]
+dynamic = ["version"]
+requires-python = ">=3.9"
+classifiers = [
+    "Intended Audience :: Developers",
+    "Operating System :: Microsoft :: Windows",
+    "Operating System :: POSIX :: Linux",
+    "Programming Language :: Python :: 3",
+    "Topic :: Scientific/Engineering :: Image Processing",
+]
+dependencies = ["numpy"]
+
+[project.urls]
+Homepage = "https://github.com/timminator/Fast-SSIM"
+documentation = "https://github.com/timminator/Fast-SSIM/blob/master/README.md"
+Repository = "https://github.com/timminator/Fast-SSIM.git"
+Issues = "https://github.com/timminator/Fast-SSIM/issues"
+
+[tool.setuptools]
+packages = ["fast_ssim"]
+
+[tool.setuptools.package-data]
+fast_ssim = ["resources/*ssim.*"]
+
+[tool.setuptools.dynamic]
+version = {attr = "fast_ssim._core.__version__"}

fast_ssim-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+