replace cluster package with KMeansColorQuantization2D in color packa…

…ge (#105)

* replace cluster package with KMeansColorQuantization2D in color package

* add functional forms of augment

* nit

* nits

README.md

@@ -104,13 +104,7 @@ net = sk.tree_unmask(net)
 | --------- | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
 | Filter    | - `{FFT,_}{Avg,Box,Gaussian,Motion}Blur2D` <br> - `{JointBilateral,Bilateral,Median}Blur2D` <br> - `{FFT,_}{UnsharpMask}2D` <br> - `{FFT,_}{Sobel,Laplacian}2D` <br> - `{FFT,_}BlurPool2D`                       |
 | Augment   | - `Adjust{Sigmoid,Log}2D` <br> - `{Adjust,Random}{Brightness,Contrast,Hue,Saturation}2D`, <br> - `RandomJigSaw2D`,`PixelShuffle2D`, <br> - `Pixelate2D`,`Posterize2D`,`Solarize2D` <br> - `FourierDomainAdapt2D` |
-| Geometric | - `{Random,_}{Horizontal,Vertical}{Translate,Flip,Shear}2D` <br> - `{Random,_}{Rotate}2D` <br> - `RandomPerspective2D` <br> - `{Random,_}WaveTransform2D` <br> - `{FFT,_}ElasticTransform2D`                     |
+| Geometric | - `{Random,_}{Horizontal,Vertical}{Translate,Flip,Shear}2D` <br> - `{Random,_}{Rotate}2D` <br> - `RandomPerspective2D` <br> - `{FFT,_}ElasticTransform2D`                     |
 | Color     | - `RGBToGrayscale2D` , `GrayscaleToRGB2D` <br> - `RGBToHSV2D`, `HSVToRGB2D`                                                                                                                                      |
 
-#### 🌈 Cluster package: `serket.cluster`
-
-| Group      | Layers     |
-| ---------- | ---------- |
-| Clustering | - `KMeans` |
-
 </details>

docs/API/augment.rst

@@ -3,25 +3,34 @@ Augment API
 .. currentmodule:: serket.image
 
 .. autoclass:: AdjustBrightness2D
-.. autoclass:: RandomBrightness2D
-
 .. autoclass:: AdjustContrast2D
-.. autoclass:: RandomContrast2D
-
 .. autoclass:: AdjustHue2D
-.. autoclass:: RandomHue2D
-
 .. autoclass:: AdjustLog2D
-
 .. autoclass:: AdjustSaturation2D
-.. autoclass:: RandomSaturation2D
-
 .. autoclass:: AdjustSigmoid2D
-.. autoclass:: RandomJigSaw2D
+.. autoclass:: FourierDomainAdapt2D
 .. autoclass:: Pixelate2D
 .. autoclass:: PixelShuffle2D
 .. autoclass:: Posterize2D
-
+.. autoclass:: RandomBrightness2D
+.. autoclass:: RandomContrast2D
+.. autoclass:: RandomHue2D
+.. autoclass:: RandomJigSaw2D
+.. autoclass:: RandomSaturation2D
 .. autoclass:: Solarize2D
 
-.. autoclass:: FourierDomainAdapt2D
+.. autofunction:: adjust_brightness_2d
+.. autofunction:: adjust_contrast_2d
+.. autofunction:: adjust_hue_3d
+.. autofunction:: adjust_log_2d
+.. autofunction:: adjust_sigmoid_2d
+.. autofunction:: fourier_domain_adapt_2d
+.. autofunction:: pixel_shuffle_3d
+.. autofunction:: pixelate_2d
+.. autofunction:: posterize_2d
+.. autofunction:: random_brightness_2d
+.. autofunction:: random_contrast_2d
+.. autofunction:: random_hue_3d
+.. autofunction:: random_jigsaw_2d
+.. autofunction:: random_saturation_3d
+.. autofunction:: solarize_2d

docs/API/color.rst

@@ -6,8 +6,10 @@ Color API
 .. autoclass:: GrayscaleToRGB2D
 .. autoclass:: RGBToHSV2D
 .. autoclass:: HSVToRGB2D
+.. autoclass:: KMeansColorQuantization2D
 
 .. autofunction:: rgb_to_grayscale
 .. autofunction:: grayscale_to_rgb
 .. autofunction:: rgb_to_hsv
-.. autofunction:: hsv_to_rgb
+.. autofunction:: hsv_to_rgb
+.. autofunction:: kmeans_color_quantization_2d

docs/API/geometric.rst

@@ -18,15 +18,11 @@ Geometric API
 .. autoclass:: VerticalShear2D    
 .. autoclass:: RandomVerticalShear2D
 
-
 .. autoclass:: VerticalFlip2D
 .. autoclass:: RandomVerticalFlip2D
 
 .. autoclass:: VerticalTranslate2D
 .. autoclass:: RandomVerticalTranslate2D
 
-.. autoclass:: WaveTransform2D
-.. autoclass:: RandomWaveTransform2D
-
 .. autoclass:: ElasticTransform2D
 .. autoclass:: FFTElasticTransform2D

docs/index.rst

@@ -85,7 +85,6 @@ Install from github::
 
     notebooks/layers_overview
     API/common
-    API/cluster
     API/nn
     API/image
     API/sepes

serket/__init__.py

@@ -31,11 +31,10 @@
     value_and_tree,
 )
 
+from serket import image, nn
 from serket._src.containers import RandomChoice, Sequential
 from serket._src.custom_transform import tree_eval, tree_state
 
-from . import cluster, image, nn
-
 __all__ = [
     # sepes
     # module utils