shapes.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 340,
   "metadata": {},
   "outputs": [],
   "source": [
    "from PIL import Image, ImageDraw\n",
    "from IPython.display import display\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAB4AAAAeCAIAAAC0Ujn1AAAAGklEQVR4nO3BMQEAAADCoPVPbQ0PoAAA4NEACqoAASaHRZ4AAAAASUVORK5CYII=\n",
      "text/plain": [
       "<PIL.Image.Image image mode=RGB size=30x30 at 0x7FF010130520>"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "x = Image.new(\"RGB\", (48, 48))\n",
    "x"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 387,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAJYAAACWCAIAAACzY+a1AAACS0lEQVR4nO3d3W6jMBRGUTKa93/lzEWlqhqgwuHH37HXuq9q2LExIW2WBQAAAAAAAAAAAAAAAAAAAAAAAJbl/X73HkINr94D2LWZ8PXKHXAvuWfk4CwUNff4zyykU3UNPdTLL4QDRw09sAf2MsNEDT2MLtvRolFDB51zR5HfNXR8e3cUCWnTomaN5tvxm0JR/3b83XuaqqxPX0LUJyUmPGm2qGUSnlmsxo5aJuG1LozafXczacK1ujNVwl1Vov7pPYD/ZZ6mZHEJm+i9VFlIf9kyPPZkOPblUiNhq/Xpvmnf2H07uoyacG3gj3EUTnh+ZRuja9Z2JvZ6kywrIR8okDBhZUteHgokbHKyd9OPJ7y2ltLbmT1V3hi7yoAJ12If91+i6kI6TIDzghKerxJycXpYUMJNCVXCZ3x6wpuMNOMnTbgpp0oTCcsrmTD84vSwlIQjXZwelpJwU0KV/BkfnfAmg834GRNuiqrSZJz3SMd4BP+BcRJueuxzUB1FLKRNF6eTV7L87UmriISbEqZLid65CTlIwmYJy8NP9bYzdzyCT6vSpF7CTWN/tOJ3/RPed6In+RxU/4Sbnvwrlr2uVXrbztS+EC4StgrsHbqQzvDG2FVCE66JuqdzwjNbBlG/lJmFR1wYtcp2dBks4doMM3XwhGtnombmjxtTwgq2979rJfxQQtQvEl7J/1//ljimz0z7vQiJY7rKJN9Xkjim+wz5JUKhw3pS9e/3Ch1WX4XuKDgq58YGAAAAAAAAAAAAAAAAAAAAAACAvv4BoXrPmsbIHvsAAAAASUVORK5CYII=\n",
      "text/plain": [
       "<PIL.Image.Image image mode=RGB size=150x150 at 0x7FEFD3CE5370>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAADAAAAAwCAIAAADYYG7QAAAEMUlEQVR4nO2WSU/qahzGn0IRKXjQVgYlojFikcQpxoE4ROLapVv9AO6NGxM/hSs/gG5cuDExMbpQcYhxShhEUTAoVFrFAYUWehfNPbn3eHIDiZ676W/7vu3zy/99+6SAioqKioqKypdClPvA2NiYJEnJZDIej7+/v//PQjqdbnNzc3BwMJvNchx3eXkZjUZTqZTf7+c4LhKJvL29FQqFPydUU1NzeXlJ0/TnpUKhwHHc7e3txcVFIBAIh8OCIJyfn2ez2bIGWZ6Qx+PZ3983mUzT09PJZNLj8bAs63a7HQ6HzWbTaDS/7Od5PpVKhUKhtbW1xcXFUiLIsoRYljWZTKlUanl5mef5lZUVACRJUhTldrsZhhkaGrJYLAMDA2az2el0MgzDMIzH4/H7/WUFlcrMzIwsy9fX13q9niAIiqJ+u40kSaPR6PV6x8fHM5mMLMsjIyMlRpQ3oY6ODgAHBweiKGo0mqWlJavVynHc3t5eIpE4OTnJ5XLhcFiSJEmS/H7/8PCwwWDIZDL39/dlBZWEXq/f3d2VZXlhYQFAa2trPp+X/83Hx8fp6enx8XFzczOAiYkJWZZjsZhery8xpYwJVVRUKDFbW1uKn06ny+fz6+vrVVVVLpeLpmmdTtfR0ZFIJJ6engD4fD4AwWCw9C4oQ0jJk2VZEAQAnZ2dAHien5ycfHx8rK2tNZlMU1NT8/PziURCEASCIJSCiMVikiR9vVBbWxtN0xzHnZycAHA6nQAikcjj4yOAdDqdTqeVEd7d3QGgKKq/vx/A0dFR6Sm/Nsd/oBj8LLre3l4ANzc3/9zT1NQE4PDwEIDBYFA+w3g8/i1CPT09APb397PZLID6+noA0Wi0qamptrYWQE1Njcvl+mnQ1dVlsVgEQQgGg6WnlHpkJEk2NjYCEARBlmWaph0OB4C5ubnZ2VlBECKRyMvLC8MwAE5PTwHQNE0QhCiKyp37YiGtVtvW1gZgc3MTQHV19cPDA8/zLMtWVlbW19crAwMgimIulwMwOjoKIBqN5vP5rxdyOp1ms1kUxWQyqcR0d3cTBMGyrF6v7+rqcjgcAwMDSk9eXV1pNBplWtFoVPH7YqG6ujqz2fz8/FxdXe31es/OznK5nCRJoVAIf5+RAkVRxWKxoqKir68PwNnZWek2ZQgpb//x48fq6iqAeDyeyWT29vYeHh62t7d5ng+FQtlsVpIk5cqbTCaDwQAgEol8i9DT09PKyorb7bbZbAzDKBXQ3t6urBaLxVQqlUgkQqFQOBwOBAJ2u91ms72+vobD4bKEyvsfUqqlvb2dpmmWZT0eT2tra0NDg9Vq1Wq1n/cLgtDS0qI057cIfUar1RqNRpfLZbVavV6vzWZrbm5uaWmxWq0URe3s7Ph8PlEU/5zQbzEYDE6n0263kyS5sbHxHREqKioqKioqpfIXDtTmj/ZK8eYAAAAASUVORK5CYII=\n",
      "text/plain": [
       "<PIL.Image.Image image mode=RGB size=48x48 at 0x7FF01018C550>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "scale = 5\n",
    "line_width = scale*2\n",
    "size = scale * 30\n",
    "radius = scale * 8\n",
    "\n",
    "x = Image.new(\"RGB\", (size, size))\n",
    "d = ImageDraw.Draw(x)\n",
    "\n",
    "center_x = size / 2\n",
    "center_y = size / 2\n",
    "white = (255, 255, 255)\n",
    "black = (0, 0, 0)\n",
    "\n",
    "def rect(d, center_x, center_y, radius, angle, fill):\n",
    "    point1x = center_x + radius * np.cos(angle)\n",
    "    point1y = center_y + radius * np.sin(angle)\n",
    "    point2x = center_x + radius * np.cos(angle+np.pi/2)\n",
    "    point2y = center_y + radius * np.sin(angle+np.pi/2)\n",
    "    point3x = center_x + radius * np.cos(angle+np.pi)\n",
    "    point3y = center_y + radius * np.sin(angle+np.pi)\n",
    "    point4x = center_x + radius * np.cos(angle-np.pi/2)\n",
    "    point4y = center_y + radius * np.sin(angle-np.pi/2)\n",
    "    \n",
    "    d.polygon([(point1x, point1y), (point2x,point2y), (point3x,point3y), (point4x,point4y)], fill=fill)\n",
    "\n",
    "def tri(d, center_x, center_y, radius, angle, fill):\n",
    "    point1x = center_x + radius * np.cos(angle)\n",
    "    point1y = center_y + radius * np.sin(angle)\n",
    "    point2x = center_x + radius * np.cos(angle+2*np.pi/3)\n",
    "    point2y = center_y + radius * np.sin(angle+2*np.pi/3)\n",
    "    point3x = center_x + radius * np.cos(angle-2*np.pi/3)\n",
    "    point3y = center_y + radius * np.sin(angle-2*np.pi/3)\n",
    "    \n",
    "    d.polygon([(point1x, point1y), (point2x,point2y), (point3x,point3y)], fill=fill)\n",
    "    \n",
    "angle = np.random.uniform(-np.pi, np.pi)\n",
    "\n",
    "rect(d, center_x, center_y, radius, angle, white)\n",
    "rect(d, center_x, center_y, radius-line_width, angle, black)\n",
    "rect(d, center_x, center_y, radius-line_width*2-1, angle, white)\n",
    "rect(d, center_x, center_y, radius-line_width*3-1, angle, black)\n",
    "\n",
    "# tri(d, center_x, center_y, radius, angle, white)\n",
    "# tri(d, center_x, center_y, radius-line_width, angle, black)\n",
    "# tri(d, center_x, center_y, radius-line_width*2-1, angle, white)\n",
    "# tri(d, center_x, center_y, radius-line_width*3-1, angle, black)\n",
    "\n",
    "display(x)\n",
    "\n",
    "small = x.resize((48, 48))\n",
    "display(small)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 338,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAJYAAACWCAIAAACzY+a1AAATf0lEQVR4nNWdv4oluRXGP3cPBQMXBiYaWJhoogWnBoOjBb+Bn8GRn8SPZMNGjvwAjjbaaKIGQ4OhoLGDulX36Pz9JNW9Mz40xdEpSSXpp++oqqe9/s1//4abraN+HuQdMu4We4P+ExdzdYNlhVOKlPN0i8i7vb4MJnfz+kmFvJj3mVvd6pR+B+YDEuST7qHXTx7HGA+V6WSs+azxzyBB9g36yWn1GP2Rgot6LiuX69C3UGPdMcEpCeKGEP0Uk2fxd92aA0Xy1qw9AGQ30YmzMA8ySzx28s3Dm2V8P5A2Xo917iycz6JRW6ZI3hqodk5fAy9dpTSj1xn0U0yewt9NKjDwlrTn3mqDVvY+cBz25dLRs/BbZdGyf1nhjodi8tTk1tiBUUzj/mfhWVk0b4tvhc1ary57QdrXGXSSc3vreygdl8WZXPrNrBx6b/HxZ2FkDzgCvzu716F457NwQJFuMQk+zpbOa9lJFOw7FL+Ds3A4i9oKwwtb2FjL8tldae07PwuZogzOIOlrda7wGQmq59Ygz/sdKc/YVoiKSXDSvv1R2nU0ZvGJ35GWIssHc8rhd2a2HLPJPJ5LUBX9+Nn/XnjWWZgEGVQkzqzCVOOqVZcEVVG39c7CJfDdzk+Hp4rzqjpfl2f1yBwbbrFxxFnIQ7LPDfsP6nedhbbhQP48Z9mHR1aOeCyjqrNQNYko5nh64UWjlUFejuQOGLG7dNpJThXlWXjiERgFmbgsPjJ/3pFrV9IgU6gq3umjwh1bObyE5UBmSkZ1QlJlGJCdjKXQW/E+HxWMkxTnMxMTHOxx7AHkkUAKsekh/ajIj8C889I5V4LK7iW+riflwHrnEC5o+lFhbzHBbyJBJhu5TTpuTMqRZNYrxK6PCreHov/AeaQE7yI+8qm9QmT2uHbm/qXiu5IgExx+RNYFv2W69iArxOpfKnJgWc8PlyCz4/usbMPvSianDwox+Kj4JomU2ZFkhUj6g1Z2xwhxZj9mQmw/Kr6TRBphI3kn8XJUrN3peOBx3pzqtzPfZyJ1KydDeoSN7dBkhuTcvv9E6lo+u7vYQHohjwfm/Pi/TqSkEN3hkSNJgn37Ip9qvk95Of6/J1K3WnTXxs+xB+xTt4l1lrlEOibByUSaCzHpoUN5Y12cm04TOer4RCKNOl+Mkw2gP4WyCeYUbJExm+WUdBp1eyuOJtIEnvu4+RTKpCjXsZVrG96/rjO8W6niaCJV/ZwyHXTeiu4iDk5xZZ5Bjuw0figSKS/BfBa9m9JaMkG31TlaLFsy8+/NPO6tqAhgeXp7xttz2/B+EswHRm7N3me5No4T3KJIh8k8SOtkm/eqQgek9Hk52rsnUizmEjhuEeXdvP2JFMfmfFvf5i/YZuVoG+rHTcwrKuYOU2Tt8RSp4tPa/te3srzKSLBc1qgYzc69RTpqSPZuYbbqIynaoL8EVxXWIKU/Bo+ZSG8d5dihKnP59YnSUhzbwjbIz7xxntR/AC8EuQSrkxONts6SDh6mcjQ1BKOyvVk5utWo27m0SYozwcZ5giHHguyVY4KTvyab0vVtEe2t2hL45fYZoNhVB5CvMyVIf/BdWYQZbX7NHeXbojsFyhgVu/uI2bm2Q37nokmkSEGGB2RONJnCMD/bufJtsYxT5jZOHjw/53Lmi5NIbSQEmW+7cgq9c0n2YglvmYRX9mUHoeLDmadS4fHMtcql7AHJT2F+L8pbKn4UTyNnzfYepYV7qrBLgsUBqUZeTqF3LrZtlxDvaDlI17Fzm1PhMMg+OaopnKVC5UfwRrgu1Y9bWT3PdexkkhXJKjvfhXku7QBZTsEd7SkqPCJ2taP110YCd7srQUazYupo5x12EnIIKpIUpb9RfH7bQ6qejKwmLoNyHPaudZI55CbncIJFKoRxumSXFIEN4WaruTkLUt2LGLiEFM6Eoh2TnqN3tzBXf1bXrszRFhMV5mknKt6cNzwDeEdK8IjwIG9yhEcIFaoILVrHDlGNG+buauIFXUUor7Z4xV5+BcUN3jaV5nWGPxdXr6h8fUCqIXXNxa3vrhXSoLylIkHB64JUoStEtxhdHX5veJb8sKkwkaCNkCo8fCevRrosr25zd9AQeyqSZi0+1TjZFKqChSf9mhMjvsN5B2+yZYQHuTk+SJVO7esM2soJRXfQKq6GCDEG35bARyxENzMoH15RXXXRhbfZ7XUG3m7O33ESru6Z1RyQCYkIavk6owYNczcZYmYumKTyJD+dOeXY0Y53TV5n3OBwOmXz6omJVBorO2k6iQVJtVeLMFdXdrX4tAptsoHH0pWdKo6DjBIpRB3XyQenJqCeCPFoh64L0hJK4EX8xsXnqDDKQyXLRIVolwglyFKOrrm3kuxKYTvM6sPmRrdVwi8id7sy8OSo36lysq1thBSlC1WDzOUYpVOkgOWgI/H5thjfzXvlDypHXa/8SHib37zOwMPmBs/KqBLkLa8ez7A4eynaakp8lBAtReXn/CgJDsDb7B0vQTd4Ska9Om5elc9O1GaTpx3ZElzlIHTyRLDiJUW3mnIg4R1D7qXofFS4S6Hi98ioDUiV90ohJmOVNY8rlVGHKSIoOl1J8Q1IcJUqLLHBY8yzpPjta9sckPNmN4RynNqH73JC4Jf8xuGFidSN9qrQRpjTMXKOhhpk+ebiWiTBUHxoAUAtfZw8WYQMvER2ync+KkDgtOSO4JgQ4VXDDlK/6XSZBKHgKVJ+4yRhDiDEG55dWl3JU5r+qICZUa5CN8izRBq83Z3Jq66WlAoLkEiZ5RRvvVvxDR+Bsqg/KhDgSW51CREeyzKpXuuXILdm28/rHrzsfiQnvyP+51IiJOG5wEqK71Q0YabudgkRZtMjpWWb3IIRSFX7IijafCbhhXuCwWb5OeKL4E3qb7Pw096dXXQ3EaKNRMxyCerKzy1Ft81BcXMW4Viot05UKNFcxK8bXqm/pBh+2rvNJvMqzFIztEJ7xrLJMelFZlHJr8ilET9JzrK8NTky5131t0X8T3s1FaaCC3vsfbXPDpBHL+o43MhZfpf8vdRiu3j5UwuRhDf8/mkj/qd90v6s1JockznLda/T5NFn4P3+78mHXXaQB0XVSyjBPHlaFd74SXhM5iTJJSD9T3s5m6ilW6fUIiZy6WrGI3EuB0ipxVehtsNfhKNB2hR6EeQUvwsP73TxHUH/0z5vdm52HT4aV7N7DpC3Nx0pRLRalJhDipafgneR7yykBF0HZu4RM2XOp700MrvCrKZ7N5cjb2u7S2ReXXeKi5SjOyYp01c11YRfg/AQ3xpgc0EiQBgN012Bw5xP+7I9uAQLhHKExy8hqgihJafSqZTjFeRivu6xL7nPTwL76CK08OaTp7tQUfC4pT/tlUU5lhRuwnVMf+ileICEAHl0tLavmQuwqteWj+a6vOG9hcfzS/wokscHVXgYKUe0u9+9mz/LJk/EFGFfFt/jvfPyInJpcxYe/I6fqwT/44nPPf9cx05/QHbKBlXIPEz1kAuXEWVOEQIeBKmbrJ49kBvCl4PiYuBdEb7h/SvwWpHr4pcsC7kgmFehNP7dRzXhn7KaPbF4Dgy/Q6CXZywfBEjsFL8eI9qYfdp/Pr7hwwHvdRSeO80y8TBWqBCEEHsfKbvlk6d6SgQSBiFakNfflb7Hx+NNB8Ar8HHzNwl+Br5sCP+NZ4afmguTLcvpk9UKFfKPyY0/Mt1qSfPFFNUteEl1AV6Ajx/w8cP+EfkC/BPABfgC/Ah8+Tc+vAA2eZKay+c4uaqyea3Cw3g55o8cbs4kaltnaR3JcntF+fQDfvgEAPgH8MuPwE//wW9/BRQ/pLKLIr0VBqxW4V0f32VMzmcGqaBegOUZn/+I3wP4+a9/x4evOzmkwMYGebp1qPD7N7XoUQVZc/t5Af4F/PrHv+CnD19+xhfzO+zDolQ0k6ImrUOFZ9kpm4Y/ZiJs654nvwJfcQF+Av6EP+MX4Jd/4tPr7WNQsVQiRlAkb/EWdVL87cwDrOtxzIew+6Kh+L0KfsBn4Efgd8AnfAH+AABff8HXX6+fF0sLEq00GaL2DYsxVTlaqDMT6UxXAy8CpdrgKU/yW3EBPu2fEJ+BCy47UAALXr7iZZfj8QaESpT2JVkNsgTZ9bnsJ9JzhTj/IpB8Zlk/gbfe4MlP+M/bJzywfxZ+Er8Wf8HLC17XK0X5yzhF8bhGX65ynHnW7Xopm1LhQNuB30cw/CJ4aOGtVzTH76w/iZ8LsDRkj7VfsL7i5RXLevstjxViCZKhmPNzk3P4OjOMlmkY1enNltax5CRC8TWofwV6U9dG8VV8WOAGcn29rr7+g6ediryiRZWLDym/JLUOqvCUj6RJzUHQslfzI7/m3X+F2PnJKvZA20CuWosS5ypgLDHFkqh8srtW263TVDhAy433pk20yoPHT8gm+Vfc3Za2ohzBsv+OdaO4aopr68CIstdcXapTtluFM3nSvZVHmMyJOH+K5bV/P6H+rtCjuF7fcpwxrcDqgLQUj0Y2tUpTtFzxucETVNgFzA3OJ0/Fb79l/3gw4bcn0qP6a8sSZgT73wCs6zWwtJwmJSgtSqdL+edPysqa5yZMVOQQUGzB2D/7DPgdSy3JbT/H372h/cPUvcK63sYakeNZWkW6ctz6zBLpMLDoVoINhpYb7IRnKV4My4Cfau3OZG20iOWWV2VFRogSUs7PytH5Twcl9piciYAcCniI4Sl+VoLeCkstqi4PIaIV6zaetWZGGiPHcRUO50zQ5BAjNHJU+dBmTtf3+LnbYBVyVIn0Fbr5nloHQEZyVOJrEqlsP8bMvTWmvyGEElueRV1ybRY9TPV6wDN5t1HhqufJgDxQJfwSLQ6qcFiC5+rPKC9JpBxFFYi6lIk0MrHSK3lQOU1rLRb/E1HmVhc2cORQITTLjGCxE3heFj1M3XGhlxTlTM0y5dqywYhl8WnPY7PB0/W3XyOlIOVU/uzmklvNc2xltA29NSpBygiZV2dVOJ850QEPLSoYBlEK5ZR3WNSCR5hakldzfmEincdmgzPJEwU8cAhLCATL48lSiBDthhBuJkEmSTWCiiSRfsPkCR8ezOIxCFH5MHFjCt7ajgJhu+w1dBVdrcVxyCTV4tN+PnliCJ5wEnIIIPFCjG0x63e0W42TW1l/vWFF5cD4/uvM/cihg+ISX0l+UR3bFZrVlWsmFNMIsW3hWAm4rbDur2olSOUX/1t7Nz7/ClNKkCBnHZKiuhLm8iMlqCpH/ewRBiRaP/youAc5EBRT8WEohcI0dK/GIv2tppGNHHEeM24P6sqo9S/YHpY/TYaKyKHF4wKDuWU7Ccwl555nMq/ZItr6rhDTB5EZNfyoYILnwQMhPqQpVPYTBeWDVM+VJYnMmrobvc5ERyxYkHBfZybJYRweDC17PSuXol0wW9xjq7dmQXVt7sHpAlNNbGUvix9O9lFxSgpNg5EgyhQKjw2TS21Dwlx40UuEKyYYbEQijfIqWpDUp30eiUBWRHNs0TUSYplL4VVTjgi4x87a1nHNJZ1IkAmmIIvXmcnDDxk8CAy22JtIZTFPp2oMlSmKdmJul/lxGKXT/AUqAOm/znSx7IEHsfplkeEH4ZNUIi22viWnlsA+zZK2EpSVk0TqHYfWuaqwfJ2Zgdf6NnFZVDYY1RkT4qgWZV33FOyVIHr4xXJE8jrDp1BZ7IR3OCUnebUVDr8HSVl5IWgdRkqQpBjV9Jz6dWbgOPQyp3WY5GmvOcijmODsge2m0KQmDMto6aMi0jpwnOx3pNPvMqT4VJHnV6qwTKScMY0SFbqOxYZAlGgrG+dp8y2elSsGvlpc6aj0iLbI8LPduip8uLkDiRz+mswbQPk6M/AuQ4gPwbjyK+IlUb6M8DinBZq8xSBOnuXVbYtWhdgFNKc8pOKDB0DF8yujwohZklGHVOs2sipU8d4ZJ22F0/c6E9+Kxp4791MhQ6WHnKq4enftiwxSJZ2rwmNguRCDW5EgcudEFdqR2PH0m1VsVMct2p12HxU2ifQwMovGzErdRMXkGhXtE12QPSzHNNyVicbm7cWfRuEhXkEVPEuF0WwiOVoIHMUxeEwm6tqxpYJ355ZI+SxKiy9x7pNT9Ormi91vZX/uBp6ZsW1iHOe7MC0y+43fhGfPxvFt0Y0QlnczsIGjDjvn/UTDs0POx85swmTUYxRdorboRlIr+Vmf5OfizGu2TpNIpZ9mznzvk5OIRqqsiyKCcbr90xRP5wcvyKyNq0J0HHsIFmggiXQNvGyOYJCJH0WIKvP8BtJNlM6QJlIzQHLX25rR88k9yaxENKrE90z/n68Zewy/nhn734WxyBDEXSfPAmNzQlpkaBHKK58T+SfOtexhd66JtB1OMtJSiPeeUynBfGDWaKLD/FTN/IH9c33iMmcy6nP55TZA0Y6ZfBZnA5O2wbJCVMSuwj1Gis8d4yQ/d+Dl/GyRxMkZL7uk7XDSIYvAkwevHHs0j2F+/MzKbWktH2qn5QpndlrXLN3+2+JTUBep+MaEyPOzNqPI0oZw8srPZxb1nG/R/kSa+wzIrvmVUJm9mjs91tUomrcNkuIjEqlqyrAsNbcYx608nGjcJvZubv04e5WnigOJlCg+pcByZvlUxvLq/dJpJ+YZ5al471ZhUsw9Eynp5HK0twbS6Uk2sBnunWLa+P8AvvyCeutIgjAAAAAASUVORK5CYII=\n",
      "text/plain": [
       "<PIL.Image.Image image mode=RGB size=150x150 at 0x7FEFD3ECCCA0>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "r = np.linspace(-1, 1, size)\n",
    "coords = np.stack(np.meshgrid(r, r), axis=-1)\n",
    "\n",
    "center = np.array([0.2, 0.2])\n",
    "\n",
    "angles = np.arctan2(coords[:, :, 0] - center[0], coords[:, :, 1] - center[1])\n",
    "magnitudes = np.linalg.norm(coords - center, axis=-1)\n",
    "angles = angles\n",
    "h = angles / (2*np.pi) + 0.5\n",
    "s = np.ones(h.shape)\n",
    "v = np.ones(h.shape)\n",
    "# v = np.clip(magnitudes * 2, 0, 1)**0.5\n",
    "\n",
    "hsv = np.stack([h, s, v], axis=-1)\n",
    "hsv = hsv * 255\n",
    "hsv = hsv.astype(np.uint8)\n",
    "\n",
    "i = Image.fromarray(hsv, mode=\"HSV\")\n",
    "i = i.convert(mode=\"RGB\")\n",
    "display(i)\n",
    "\n",
    "rainbow = np.asarray(i)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 388,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAADAAAAAwCAIAAADYYG7QAAADw0lEQVR4nO3Wu2+bVRjH8Y+vqR0TN07S1E2ThpA2aYCECqmAAHFbWJhYmICdkQUGJP4LJNSxAywdWApIqAxASxGCFKlNaAk0IVfHzqWOXV/yvgx0gM0WiVj8HY+e8/y+OjpH56FDhw4dOnTocKBE2t3wyqOagbWixS3Vxv8tlIi78oFnJ1RWbRTcWbKwZr3k6oqNittle4H98D8JxduqzhxxdpyodNZo2mg/u2xRsl+wsWGp5NeimxXz+4mS2C+aFUFVcFhC+X7Jo0S8c8FawVTORNZk0lBoMCHfI19zvko1Qoa+ot51qTmJy1YvuHnwQhMPy5y0fsen1xR3Xfq7RVQ6brJbX9RzcQNVTweyghGJPuk+fVPyV223GNGe0OlxMqpR5apIRCqpUtMM7NZdr8PlB03DLrvT6v0KF73QI3ld4VCEpqfJub6g0RSN+ORdx1I27ro2b7ng51W1uvmqJk3hVdXndaf07AhX7R28UFeXsSkyinVB6MywV1+QqDPmtWn+ZEntT/O/C9YjrzcsCI9LJ2R3RBftHLxQIunUFHFf//DAL9GvXvDlrIfqTkfkUhJHTGcsFx5cmZcM0XPL6n7LD60NoWxOI6kZKu7CzAxDipve/MjWPf1pmbi3jvnwIcuRsESEnB5yd91tHobQ2FmpnLkNP/0MI+MMulO2dQ82KzYZewxWopAWf8oj9P5opfWUNoT6RmyzUlGtwuRTsLD9r5rRPGU/xCAlnnac/KJ7radEWy8dfVKZb793vwJdJ2xxY9GpUf390Jt2eoZei+AJwwNmtmRuWWo9pdUTisalTymwVhKGMjnBkN944wNvv6dRUritsWqglz2zIfTJR5zbE27aPQShmJ6zNvjxCqSP+qNguejkhMwR2RNGT8gRBBrUP4bnvVY2fMW1uvrBC2VH3M9abSitwcaC988RMTgh2WX4CQNDRp82POh+w+1dRGeNXOJzvzXUDl4onVfLKu6KHHX8GcUb9muCpvU5WJr9ZykhEl9Ez88HVt1o3aYNod7zylR6vPyZBtuLKjvWr9kr2PxGrag2J6zQpAJSmcVzqZU5wdZt7UxIrQrtbZu9JDkpMSjsE45o0P04BBQD9fV0uDxo7qT5bjejGg+bGWwmyr6b185g2d7EGE2JpaUfF8uJTohMuX9GdTjWPHZUbIhxpjnPi26mfMP3JRfH1bdaj2jvtw+qgqqdr/+xFKM7cLri2LJnkgZjxrqNnxQ+Yjjtp1ua5bYi2h7yWyKVkh+RPa4aN/fVoUR06NChQ4cOHVrkL1Ptae2s8SecAAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<PIL.Image.Image image mode=RGB size=48x48 at 0x7FEFD3F25790>"
      ]
     },
     "execution_count": 388,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "x.readonly = False\n",
    "tri = np.asarray(x).astype(np.float32) / 255.0\n",
    "rbow = np.asarray(rainbow).astype(np.float32) / 255.0\n",
    "tri *= rbow\n",
    "colorful = tri\n",
    "\n",
    "colorful = colorful * 255\n",
    "colorful = colorful.astype(np.uint8)\n",
    "\n",
    "r = Image.fromarray(colorful, mode=\"RGB\")\n",
    "r.resize((48, 48))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 342,
   "metadata": {},
   "outputs": [
    {
     "ename": "AttributeError",
     "evalue": "'ImageDraw' object has no attribute 'classes'",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-342-3a0d962a20ab>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0;32mfor\u001b[0m \u001b[0mi\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mclss\u001b[0m \u001b[0;32min\u001b[0m \u001b[0md\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mclasses\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      2\u001b[0m     \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mi\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mclss\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mAttributeError\u001b[0m: 'ImageDraw' object has no attribute 'classes'"
     ]
    }
   ],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3.8.3 64-bit ('env': venv)",
   "language": "python",
   "name": "python38364bitenvvenv0a5e483af100464fabc3d15b325cd6be"
  },
  "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.8.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}