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))