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