Helper functions¶
-
tensorboardX.utils.figure_to_image(figures, close=True)[source]¶ Render matplotlib figure to numpy format.
Note that this requires the
matplotlibpackage.Parameters: - figure (matplotlib.pyplot.figure) – figure or a list of figures
- close (bool) – Flag to automatically close the figure
Returns: image in [CHW] order
Return type: numpy.array