added clustering

nb.ipynb

@@ -137,13 +137,13 @@
     "# def normalize(color):\n",
     "#     return np.array([color[0] / 255, color[1] / 255, color[2] / 255], dtype=np.float32)\n",
     "\n",
-    "def sort_by_brightness(palette):\n",
+    "def sort_by_brightness(palette: np.uint8):\n",
     "    # https://stackoverflow.com/a/596241\n",
     "    luminosity = (0.2126 * palette[:, 2] + 0.7152 * palette[:, 1] + 0.0722 * palette[:, 0])\n",
     "\n",
     "    return palette[np.argsort(luminosity)]\n",
     "\n",
-    "def display_palette(palette, sort=False):\n",
+    "def display_palette(palette: np.uint8, sort=False):\n",
     "    swatch_size = 50\n",
     "    num_colors = palette.shape[0]\n",
     "    palette_image = np.zeros((swatch_size, swatch_size * num_colors, 3), dtype=np.uint8)\n",