elektronn3.training.handlers module

elektronn3.training.handlers.get_cmap(out_channels)

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

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

elektronn3.training.handlers.write_to_kzip(trainer, pred_batch)

Return type

None