File size: 1,574 Bytes
424188c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import cv2
import svgwrite
import colorsys
def plot_preds(image, corners, edges):
for line in edges:
cv2.line(image, tuple(line[:2]), tuple(line[2:]), (255, 255, 0), 2)
for c in corners:
cv2.circle(image, (int(c[0]), int(c[1])), 3, (0, 0, 255), -1)
return image
def random_colors(N, bright=True, same=False, colors=None):
brightness = 1.0 if bright else 0.7
if colors is None or same:
if same:
hsv = [(0, 1, brightness) for i in range(N)]
else:
hsv = [(i / N, 1, brightness) for i in range(N)]
else:
hsv = [(colors[i], 1, brightness) for i in range(N)]
colors = list(map(lambda c: colorsys.hsv_to_rgb(*c), hsv))
return colors
def svg_generate(image_link, corners, edges, name, size=512):
dwg = svgwrite.Drawing(name + '.svg', size=('{}'.format(size), '{}'.format(size)))
shapes = dwg.add(dwg.g(id='shape', fill='black'))
# colors = random_colors(len(edges), same=True)
shapes.add(dwg.image(href=image_link, size=(size, size)))
scale = size / 256
for i, edge in enumerate(edges):
x = edge[:2] * scale
y = edge[2:] * scale
shapes.add(dwg.line((int(x[0]), int(x[1])), (int(y[0]), int(y[1])),
stroke="#EE6507", stroke_width=3*scale, opacity=0.7))
for i, corner in enumerate(corners):
shapes.add(dwg.circle((int(corners[i][0] * scale), int(corners[i][1]) * scale), r=4*scale,
stroke='green', fill='white', stroke_width=2*scale, opacity=0.8))
return dwg
|