elektronn3.training.handlers module

elektronn3.training.handlers.get_cmap(out_channels)[source]
elektronn3.training.handlers.plot_image(image, overlay=None, overlay_alpha=0.5, cmap=None, out_channels=None, colorbar=True, filename=None, vmin=None, vmax=None)[source]

Plots a 2D image to a malplotlib figure.

For gray-scale images, use cmap='gray'. For label matrices (segmentation targets or class predictions), specify the global number of possible classes in out_channels.

Return type

Figure