def add_figure(self, tag, figure, step, walltime=None): """Add an figure to vdl record file. Args: tag (string): Data identifier figure (matplotlib.figure.Figure): Image represented by a Figure step (int): Step of image walltime (int): Wall time of image dataformats (string): Format of image Example: form matplotlib import pyplot as plt import numpy as np x = np.arange(100) y = x ** 2 + 1 plt.plot(x, y) fig = plt.gcf() writer.add_figure(tag="lll", figure=fig, step=0) """ if '%' in tag: raise RuntimeError("% can't appear in tag!") walltime = round(time.time() * 1000) if walltime is None else walltime img = figure_to_image(figure) self._get_file_writer().add_record( image(tag=tag, image_array=img, step=step, walltime=walltime))
def add_image(self, tag, img, step, walltime=None, dataformats="HWC"): """Add an image to vdl record file. Args: tag (string): Data identifier img (np.ndarray): Image represented by a numpy.array step (int): Step of image walltime (int): Wall time of image dataformats (string): Format of image Example: from PIL import Image import numpy as np I = Image.open("./test.png") I_array = np.array(I) writer.add_image(tag="lll", img=I_array, step=0) """ if '%' in tag: raise RuntimeError("% can't appear in tag!") walltime = round(time.time() * 1000) if walltime is None else walltime self._get_file_writer().add_record( image(tag=tag, image_array=img, step=step, walltime=walltime, dataformats=dataformats))