Spaces:

mega-snowman
/

image-to-color-wheel

Running

App Files Files Community

T.Masuda commited on Jun 27

Commit

1e2f65e

•

1 Parent(s): bb2a853

create app

Browse files

Files changed (6) hide show

app.py +96 -0
color.py +36 -0
color_wheel.py +146 -0
examples/example1.jpg +0 -0
examples/example2.jpg +0 -0
requirements.txt +2 -0

app.py ADDED Viewed

	@@ -0,0 +1,96 @@

+import gradio as gr
+from PIL import ImageEnhance
+from color import Color
+from color_wheel import ColorWheel
+def create_color_table(color):
+	sb = []
+	sb.append('''<!DOCTYPE html>
+<html lang="ja" xmlns="http://www.w3.org/1999/xhtml">
+  <head>
+	<meta charset="utf-8" />
+	<title></title>
+	<style>
+	  .container {display:grid;grid-template-columns:64px 64px 64px 64px 64px 64px 64px 64px 64px 64px 64px 64px}
+	  .container > div {padding:3px; color:white; font-size:9pt; height: 32px}
+	</style>
+  </head>
+  <body>
+''')
+	cwt = ColorWheel(color)
+	colors = [[], [], [], [], [], [], [], [], [], [], [], [], [], [], []]
+	for h in cwt.hue:
+		cw = ColorWheel(Color.from_rgb(h.r, h.g, h.b))
+		for i, t in enumerate(cw.tone15):
+			colors[i].append(t)
+	for hue in colors:
+		sb.append('	<div class="container">\n')
+		for h in hue:
+			sb.append('	  <div style="background-color:rgb({},{},{})">#{:02x}{:02x}{:02x}</div>\n'.format(h.r, h.g, h.b, h.r, h.g, h.b))
+		sb.append('	</div>\n')
+	sb.append('''  </body >
+</html>''')
+	html = ''.join(sb)
+	return html
+def chunks(list, n):
+	c = []
+	for i in range(0, len(list), n):
+		c.append(list[i:i + n])
+	return c
+def calc_score(colors, colorset):
+	s = []
+	srccolors = colors.copy()[0:32]
+	for c in colorset:
+		cd = []
+		for color in srccolors:
+			cd.append((color.r - c.r) ** 2 + (color.g - c.g) ** 2 + (color.b - c.b) ** 2)
+		i = min(range(len(cd)), key=cd.__getitem__)
+		s.append(cd[i])
+		srccolors.pop(i)
+	return sum(s)
+def handv(c):
+	cw = ColorWheel.rgb_to_hsv(c.r / 255.0, c.g / 255.0, c.b / 255.0)
+	return cw[1] + cw[2]
+def process_image(image, sf=0.0, cf=0.0):
+	if image is None:
+		return None
+	e = ImageEnhance.Color(image)
+	img = e.enhance(cf + 1.0)
+	imglim = img.quantize(256)
+	palette = list(chunks(imglim.getpalette(), 3))
+	colors = []
+	for p in palette:
+		r, g, b = p
+		color = Color(255, r, g, b)
+		colors.append(color)
+	colors.sort(key=handv, reverse=True)
+	scores = []
+	for c in colors[0:32 + int(sf * 96)]:
+		cw = ColorWheel(c)
+		scores.append(calc_score(colors, cw.hue))
+	i = min(range(len(scores)), key=scores.__getitem__)
+	return create_color_table(colors[i])
+app = gr.Interface(
+	title='Image To Color Wheel',
+	fn=process_image,
+	inputs=[
+		gr.Image(type='pil'),
+		gr.Slider(maximum=1.0, step=0.1, label='sampling factor', value=0.0),
+		gr.Slider(maximum=1.0, step=0.1, label='color factor', value=0.0),
+	],
+	outputs=gr.HTML(),
+	allow_flagging='never',
+	examples=[['examples/example1.jpg'], ['examples/example2.jpg']],
+	#cache_examples=False,
+	concurrency_limit=20,
+)
+app.launch()

color.py ADDED Viewed

	@@ -0,0 +1,36 @@

+class Color:
+	@property
+	def a(self):
+		return self._a
+	@property
+	def r(self):
+		return self._r
+	@property
+	def g(self):
+		return self._g
+	@property
+	def b(self):
+		return self._b
+	def __init__(self, a, r, g, b):
+		self._a = a
+		self._r = r
+		self._g = g
+		self._b = b
+	@staticmethod
+	def from_argb(a, r, g, b):
+		return Color(a, r, g, b)
+	@staticmethod
+	def from_rgb(r, g, b):
+		return Color(0xff, r, g, b)
+	def __eq__(self, other):
+		return self.a == other.a and self.r == other.r and self.g == other.g and self.b == other.b
+	def __str__(self):
+		return 'ARGB({},{},{},{})'.format(self.a, self.r, self.g, self.b)

color_wheel.py ADDED Viewed

	@@ -0,0 +1,146 @@

+import math
+from color import Color
+class ColorWheel:
+	@property
+	def base_color(self):
+		return self._base_color
+	@property
+	def hue(self):
+		return [
+			self.base_color, self.add_h(1.0 / 12), self.add_h(2.0 / 12), self.add_h(3.0 / 12), self.add_h(4.0 / 12), self.add_h(5.0 / 12),
+			self.add_h(-6.0 / 12), self.add_h(-5.0 / 12), self.add_h(-4.0 / 12), self.add_h(-3.0 / 12), self.add_h(-2.0 / 12), self.add_h(-1.0 / 12)
+		]
+	@property
+	def tone(self):
+		return [self.add_white(-2.0 / 16), self.add_white(-1.0 / 16), self.base_color, self.add_white(1.0 / 16), self.add_white(2.0 / 16)]
+	@property
+	def tone15(self):
+		return [self.add_white(-7.0 / 16), self.add_white(-6.0 / 16), self.add_white(-5.0 / 16), self.add_white(-4.0 / 16), self.add_white(-3.0 / 16),
+				self.add_white(-2.0 / 16), self.add_white(-1.0 / 16), self.base_color, self.add_white(1.0 / 16), self.add_white(2.0 / 16),
+				self.add_white(3.0 / 16), self.add_white(4.0 / 16), self.add_white(5.0 / 16), self.add_white(6.0 / 16), self.add_white(7.0 / 16)]
+	@property
+	def complementary_colors(self):
+		return [self.base_color, self.add_h(0.5)]
+	@property
+	def triadic_colors(self):
+		return [self.add_h(-4.0 / 12), self.base_color, self.add_h(4.0 / 12)]
+	@property
+	def split_complementary_colors(self):
+		return [self.add_h(-5.0 / 12), self.base_color, self.add_h(5.0 / 12)]
+	@property
+	def analogous_colors(self):
+		return [self.add_h(-2.0 / 12), self.add_h(-1.0 / 12), self.base_color, self.add_h(1.0 / 12), self.add_h(2.0 / 12)]
+	def __init__(self, c):
+		self._base_color = c
+		self._r = c.r / 255.0
+		self._g = c.g / 255.0
+		self._b = c.b / 255.0
+	@staticmethod
+	def from_hsv(h, s, v):
+		r, g, b = ColorWheel.hsv_to_rgb(h, s, v)
+		c = Color.from_rgb(round(r * 255), round(g * 255), round(b * 255))
+		return ColorWheel(c)
+	def add_white(self, value):
+		r, g, b = (self._r + value, self._g + value, self._b + value)
+		r = min(max(r, 0.0), 1.0)
+		g = min(max(g, 0.0), 1.0)
+		b = min(max(b, 0.0), 1.0)
+		return self._from_rgb(r, g, b)
+	def add_h(self, value):
+		h, s, v = ColorWheel.rgb_to_hsv(self._r, self._g, self._b)
+		h = (h + value) % 1.0
+		if h < 0.0:
+			h += 1.0
+		r, g, b = ColorWheel.hsv_to_rgb(h, s, v)
+		return self._from_rgb(r, g, b)
+	def add_s(self, value):
+		h, s, v = ColorWheel.rgb_to_hsv(self._r, self._g, self._b)
+		s += value
+		s = min(max(s, 0.0), 1.0)
+		r, g, b = ColorWheel.hsv_to_rgb(h, s, v)
+		return self._from_rgb(r, g, b)
+	def add_v(self, value):
+		h, s, v = ColorWheel.rgb_to_hsv(self._r, self._g, self._b)
+		v += value
+		v = min(max(v, 0.0), 1.0)
+		r, g, b = ColorWheel.hsv_to_rgb(h, s, v)
+		return self._from_rgb(r, g, b)
+	@staticmethod
+	def rgb_to_hsv(r, g, b):
+		if r < 0.0 or r > 1.0:
+			raise ValueError()
+		if g < 0.0 or g > 1.0:
+			raise ValueError()
+		if b < 0.0 or b > 1.0:
+			raise ValueError()
+		cmax = max(r, g, b)
+		cmin = min(r, g, b)
+		h = cmax - cmin
+		if h > 0.0:
+			if cmax == r:
+				h = (g - b) / h
+				if h < 0.0:
+					h += 6.0
+			elif cmax == g:
+				h = 2.0 + (b - r) / h
+			else:
+				h = 4.0 + (r - g) / h
+		h /= 6.0
+		s = cmax - cmin
+		if cmax != 0.0:
+			s /= cmax
+		v = cmax
+		return h, s, v
+	@staticmethod
+	def hsv_to_rgb(h, s, v):
+		if h < 0.0 or h > 1.0:
+			raise ValueError()
+		if s < 0.0 or s > 1.0:
+			raise ValueError()
+		if v < 0.0 or v > 1.0:
+			raise ValueError()
+		r = v
+		g = v
+		b = v
+		if s > 0.0:
+			h *= 6.0
+			i = math.floor(h)
+			f = h - i
+			if i == 1:
+				r *= 1 - s * f
+				b *= 1 - s
+			elif i == 2:
+				r *= 1 - s
+				b *= 1 - s * (1 - f)
+			elif i == 3:
+				r *= 1 - s
+				g *= 1 - s * f
+			elif i == 4:
+				r *= 1 - s * (1 - f)
+				g *= 1 - s
+			elif i == 5:
+				g *= 1 - s
+				b *= 1 - s * f
+			else:
+				g *= 1 - s * (1 - f)
+				b *= 1 - s
+		return r, g, b
+	def _from_rgb(self, r, g, b):
+		return Color.from_argb(self.base_color.a, round(r * 255), round(g * 255), round(b * 255))

examples/example1.jpg ADDED Viewed

examples/example2.jpg ADDED Viewed

requirements.txt ADDED Viewed

	@@ -0,0 +1,2 @@


1	+ gradio
2	+ pillow