Added simplify image

.gitignore

@@ -0,0 +1,3 @@
+__pycache__/
+.ipynb_checkpoints/
+.DS_Store

ColorByNumber.ipynb

@@ -0,0 +1,69 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "af221ec3-755c-42d5-8bb4-739644c482b6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import cv2\n",
+    "\n",
+    "from colorbynumber.utils import show_image\n",
+    "from colorbynumber.simplify_image import simplify_image"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "497a508a-87fc-49f7-baca-3505ff0b57f1",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Image from: https://www.pexels.com/photo/scarlet-macaw-1618424/\n",
+    "# Photographer: Tim Mossholder\n",
+    "\n",
+    "image = cv2.imread(\"ExampleImages/Macaw.jpeg\")\n",
+    "image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)\n",
+    "show_image(image)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7368c3bd-a7a6-467b-be90-cf628d5b3b81",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "color_list = [\n",
+    "    (255, 0, 0), # Red\n",
+    "    (0, 255, 0), # Green\n",
+    "    (0, 0, 255), # Blue\n",
+    "]\n",
+    "simplified_image = simplify_image(image, color_list)\n",
+    "show_image(simplified_image)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.18"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

colorbynumber/simplify_image.py

@@ -0,0 +1,37 @@
+import numpy as np
+
+def _closest_color(pixel, color_list):
+    """
+    Finds the closest color in the list to the given pixel (RGB values)
+    
+    Args:
+      pixel: A tuple representing the RGB values of a pixel (R, G, B).
+      color_list: A list of tuples representing RGB values of allowed colors.
+    
+    Returns:
+      A tuple representing the RGB values of the closest color in the list.
+    """
+    distances = np.array([np.linalg.norm(np.array(pixel) - np.array(color)) for color in color_list])
+    return color_list[np.argmin(distances)]
+
+def simplify_image(image, color_list):
+    """
+    Converts all colors in an image to the closest color in the color list.
+    
+    Args:
+      image: Image in the RGB color space as a 3D array.
+      color_list: A list of tuples representing RGB values of allowed colors.
+    
+    Returns:
+      A copy of the image with all colors replaced with the closest color in the list.
+    """
+    
+    # Replace each pixel with closest color from the list
+    converted_image = image.copy()  # Operate on a copy
+    height, width, channels = image.shape
+    for y in range(height):
+        for x in range(width):
+            pixel = image[y, x]
+            closest = _closest_color(pixel, color_list)
+            converted_image[y, x] = closest
+    return converted_image

colorbynumber/utils.py

@@ -0,0 +1,6 @@
+from matplotlib import pyplot as plt
+
+def show_image(image):
+    plt.imshow(image)
+    plt.axis('off')
+    plt.show()

environment.yml

@@ -7,3 +7,4 @@ dependencies:
   - python=3.9
   - opencv
   - jupyterlab
+  - matplotlib