Improved dithering for 2 color palettes

- taking square root of 2 color PMFs for more dithering - added Pal.list() to Pal - updated README.md examples and examples Notebook - increased version number to 2.0.1
sedthh · Apr 7, 2021 · 3685a4d · 3685a4d
1 parent de1ce9a
commit 3685a4d
Show file tree

Hide file tree

Showing 9 changed files with 83 additions and 11 deletions.
diff --git a/README.md b/README.md
@@ -51,7 +51,7 @@ pyx_robocop.transform(car)
 pyx_robocop.transform(robocop)
 """
 ```
-![Fit Transform Palette](/examples/p_fit_transform_.png)
+![Fit Transform Palette](/examples/p_fit_transform.png)
 
 For a single image, it is possible to call both fit() and transform() at the same time:
 

diff --git a/examples.ipynb b/examples.ipynb
diff --git a/examples/p_br.png b/examples/p_br.png
diff --git a/examples/p_br2.png b/examples/p_br2.png
diff --git a/examples/p_fit_transform_.png → examples/p_fit_transform.png b/examples/p_fit_transform_.png → examples/p_fit_transform.png
diff --git a/pyxelate/__init__.py b/pyxelate/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "2.0.0"
+__version__ = "2.0.1"
 
 __short_description__ = (
     "Downsample images to 8-bit pixel art.",

diff --git a/pyxelate/pal.py b/pyxelate/pal.py
@@ -20,6 +20,11 @@ def __next__(self):
         else:
             raise StopIteration
 
+    @classmethod
+    def list(self):
+        """list all available Palette names"""
+        return list(map(lambda x: x.name, self))
+
     def from_hex(hex_list):
         """Generate Pal palette from list of #HEX color values"""
         hex_list = [h.lstrip("#") for h in hex_list]

diff --git a/pyxelate/pyx.py b/pyxelate/pyx.py
@@ -76,6 +76,15 @@ def fit(self, X, y=None):
             warnings.warn("Pyxelate could not properly assign colors, try a different palette size for better results!", Warning)
         return self
 
+    def predict_proba(self, X):
+        p = super().predict_proba(X)
+        if self.find_palette:
+            if self.palette < 3:
+                return np.sqrt(p)
+        elif len(self.palette) < 3:
+            return np.sqrt(p)
+        return p
+
 
 class Pyx(BaseEstimator, TransformerMixin):
     """Pyx extends scikit-learn transformers"""

diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 from setuptools import setup
 
 setup(name='pyxelate',
-      version='2.0.0',
+      version='2.0.1',
       description='Pyxelate is a Python class that converts images into 8-bit pixel art.',
       url='http://github.com/sedthh/pyxelate',
       author='sedthh',