compressors 0.0.1__tar.gz

compressors-0.0.1/LICENSE

@@ -0,0 +1 @@
+Contact Dan Jacobellis

compressors-0.0.1/PKG-INFO

@@ -0,0 +1,15 @@
+Metadata-Version: 2.4
+Name: compressors
+Version: 0.0.1
+Summary: compresspors
+Author-email: Dan Jacobellis <danjacobellis@utexas.edu>
+License-Expression: LicenseRef-Proprietary
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch
+Dynamic: license-file
+
+# codecs

compressors-0.0.1/README.md

@@ -0,0 +1 @@
+# codecs

compressors-0.0.1/pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "compressors"
+version = "0.0.1"
+description = "compresspors"
+authors = [
+    {name = "Dan Jacobellis", email = "danjacobellis@utexas.edu"}
+]
+readme = "README.md"
+license = "LicenseRef-Proprietary"
+license-files = ["LICENSE"]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Operating System :: OS Independent"
+]
+dependencies = [
+    "torch",
+]
+requires-python = ">=3.6"

compressors-0.0.1/setup.cfg

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

compressors-0.0.1/src/compressors/evaluate.py

@@ -0,0 +1,46 @@
+import torch, io, datasets, PIL.Image, numpy as np
+from piq import LPIPS, DISTS, SSIMLoss
+from torchvision.transforms.v2.functional import pil_to_tensor, resize, to_pil_image
+
+def kodak(encode_fn, decode_fn, resample_factor=1.0, img_format='torch', device='cuda:0'):
+    dataset = datasets.load_dataset("danjacobellis/kodak")['validation']
+    lpips_loss, dists_loss, ssim_loss = LPIPS().to(device), DISTS().to(device), SSIMLoss().to(device)
+    
+    def evaluate_quality(sample):
+        img = sample['image']
+        orig_size = (img.height, img.width)
+        pixels = img.width * img.height
+        
+        if resample_factor != 1.0:
+            new_size = (int(img.height * resample_factor), int(img.width * resample_factor))
+            img = img.resize((new_size[1], new_size[0]), PIL.Image.LANCZOS)
+        
+        if img_format == 'torch':
+            x_in = pil_to_tensor(img).float() / 127.5 - 1.0
+            buff = encode_fn(x_in)
+            x_dec = decode_fn(buff)
+            if resample_factor != 1.0:
+                x_dec = resize(x_dec, orig_size)
+            x_hat_01 = (x_dec.to(device) / 2 + 0.5).unsqueeze(0)
+        else:
+            buff = encode_fn(img)
+            dec_img = decode_fn(buff)
+            if resample_factor != 1.0:
+                dec_img = dec_img.resize((orig_size[1], orig_size[0]), PIL.Image.LANCZOS)
+            x_hat_01 = (pil_to_tensor(dec_img).to(device).float() / 127.5 - 1.0).unsqueeze(0) / 2 + 0.5
+        
+        x_orig_01 = (pil_to_tensor(sample['image']).to(device).float() / 127.5 - 1.0).unsqueeze(0) / 2 + 0.5
+        size_bytes = len(buff.getbuffer()) if hasattr(buff, 'getbuffer') else len(buff)
+        
+        mse = torch.nn.functional.mse_loss(x_orig_01[0], x_hat_01[0])
+        return {
+            'pixels': pixels,
+            'bpp': 8 * size_bytes / pixels,
+            'PSNR': -10 * mse.log10().item(),
+            'LPIPS_dB': -10 * np.log10(lpips_loss(x_orig_01, x_hat_01).item()),
+            'DISTS_dB': -10 * np.log10(dists_loss(x_orig_01, x_hat_01).item()),
+            'SSIM': 1 - ssim_loss(x_orig_01, x_hat_01).item(),
+        }
+    
+    results = dataset.map(evaluate_quality)
+    return {m: float(np.mean(results[m])) for m in ['pixels', 'bpp', 'PSNR', 'LPIPS_dB', 'DISTS_dB', 'SSIM']}

compressors-0.0.1/src/compressors.egg-info/PKG-INFO

@@ -0,0 +1,15 @@
+Metadata-Version: 2.4
+Name: compressors
+Version: 0.0.1
+Summary: compresspors
+Author-email: Dan Jacobellis <danjacobellis@utexas.edu>
+License-Expression: LicenseRef-Proprietary
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch
+Dynamic: license-file
+
+# codecs

compressors-0.0.1/src/compressors.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+LICENSE
+README.md
+pyproject.toml
+src/compressors/evaluate.py
+src/compressors.egg-info/PKG-INFO
+src/compressors.egg-info/SOURCES.txt
+src/compressors.egg-info/dependency_links.txt
+src/compressors.egg-info/requires.txt
+src/compressors.egg-info/top_level.txt

compressors-0.0.1/src/compressors.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

compressors-0.0.1/src/compressors.egg-info/requires.txt

@@ -0,0 +1 @@
+torch

compressors-0.0.1/src/compressors.egg-info/top_level.txt

@@ -0,0 +1 @@
+compressors