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