Spaces:
Running
Running
File size: 4,529 Bytes
e17e8cc |
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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
"""This code is taken from <https://github.com/alexandre01/deepsvg>
by Alexandre Carlier, Martin Danelljan, Alexandre Alahi and Radu Timofte
from the paper >https://arxiv.org/pdf/2007.11301.pdf>
"""
import src.preprocessing.deepsvg.deepsvg_svglib.svg as svg_lib
from .geom import Bbox, Point
import math
import numpy as np
import IPython.display as ipd
from moviepy.editor import ImageClip, concatenate_videoclips, ipython_display
def make_grid(svgs, num_cols=3, grid_width=24):
"""
svgs: List[svg_lib.SVG]
"""
nb_rows = math.ceil(len(svgs) / num_cols)
grid = svg_lib.SVG([], viewbox=Bbox(grid_width * num_cols, grid_width * nb_rows))
for i, svg in enumerate(svgs):
row, col = i // num_cols, i % num_cols
svg = svg.copy().translate(Point(grid_width * col, grid_width * row))
grid.add_path_groups(svg.svg_path_groups)
return grid
def make_grid_grid(svg_grid, grid_width=24):
"""
svg_grid: List[List[svg_lib.SVG]]
"""
nb_rows = len(svg_grid)
num_cols = len(svg_grid[0])
grid = svg_lib.SVG([], viewbox=Bbox(grid_width * num_cols, grid_width * nb_rows))
for i, row in enumerate(svg_grid):
for j, svg in enumerate(row):
svg = svg.copy().translate(Point(grid_width * j, grid_width * i))
grid.add_path_groups(svg.svg_path_groups)
return grid
def make_grid_lines(svg_grid, grid_width=24):
"""
svg_grid: List[List[svg_lib.SVG]]
"""
nb_rows = len(svg_grid)
num_cols = max(len(r) for r in svg_grid)
grid = svg_lib.SVG([], viewbox=Bbox(grid_width * num_cols, grid_width * nb_rows))
for i, row in enumerate(svg_grid):
for j, svg in enumerate(row):
j_shift = (num_cols - len(row)) // 2
svg = svg.copy().translate(Point(grid_width * (j + j_shift), grid_width * i))
grid.add_path_groups(svg.svg_path_groups)
return grid
COLORS = ["aliceblue", "antiquewhite", "aqua", "aquamarine", "azure", "beige", "bisque", "black", "blanchedalmond",
"blue", "blueviolet", "brown", "burlywood", "cadetblue", "chartreuse", "chocolate", "coral", "cornflowerblue",
"cornsilk", "crimson", "cyan", "darkblue", "darkcyan", "darkgoldenrod", "darkgray", "darkgreen", "darkgrey",
"darkkhaki", "darkmagenta", "darkolivegreen", "darkorange", "darkorchid", "darkred", "darksalmon",
"darkseagreen", "darkslateblue", "darkslategray", "darkslategrey", "darkturquoise", "darkviolet", "deeppink",
"deepskyblue", "dimgray", "dimgrey", "dodgerblue", "firebrick", "floralwhite", "forestgreen", "fuchsia",
"gainsboro", "ghostwhite", "gold", "goldenrod", "gray", "green", "greenyellow", "grey", "honeydew", "hotpink",
"indianred", "indigo", "ivory", "khaki", "lavender", "lavenderblush", "lawngreen", "lemonchiffon",
"lightblue", "lightcoral", "lightcyan", "lightgoldenrodyellow", "lightgray", "lightgreen", "lightgrey",
"lightpink", "lightsalmon", "lightseagreen", "lightskyblue", "lightslategray", "lightslategrey",
"lightsteelblue", "lightyellow", "lime", "limegreen", "linen", "magenta", "maroon", "mediumaquamarine",
"mediumblue", "mediumorchid", "mediumpurple", "mediumseagreen", "mediumslateblue", "mediumspringgreen",
"mediumturquoise", "mediumvioletred", "midnightblue", "mintcream", "mistyrose", "moccasin", "navajowhite",
"navy", "oldlace", "olive", "olivedrab", "orange", "orangered", "orchid", "palegoldenrod", "palegreen",
"paleturquoise", "palevioletred", "papayawhip", "peachpuff", "peru", "pink", "plum", "powderblue", "purple",
"red", "rosybrown", "royalblue", "saddlebrown", "salmon", "sandybrown", "seagreen", "seashell", "sienna",
"silver", "skyblue", "slateblue", "slategray", "slategrey", "snow", "springgreen", "steelblue", "tan", "teal",
"thistle", "tomato", "turquoise", "violet", "wheat", "white", "whitesmoke", "yellow", "yellowgreen"]
def to_gif(img_list, file_path=None, frame_duration=0.1, do_display=True):
clips = [ImageClip(np.array(img)).set_duration(frame_duration) for img in img_list]
clip = concatenate_videoclips(clips, method="compose", bg_color=(255, 255, 255))
if file_path is not None:
clip.write_gif(file_path, fps=24, verbose=False, logger=None)
if do_display:
src = clip if file_path is None else file_path
ipd.display(ipython_display(src, fps=24, rd_kwargs=dict(logger=None), autoplay=1, loop=1))
|