Helper functions¶
-
tensorboardX.utils.
figure_to_image
(figures, close=True)[source]¶ Render matplotlib figure to numpy format.
Note that this requires the
matplotlib
package.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