Plot overlay images to TensorBoard

ELEKTRONN / elektronn3

A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data

MIT License

160 stars 27 forks source link

Plot overlay images to TensorBoard #8

Closed mdraw closed 6 years ago

mdraw commented 6 years ago

Prediction/target overlay images should be plotted to TensorBoard in addition to pure prediction and target images. Existing code (for matplotlib, needs to be adapted): https://github.com/ELEKTRONN/elektronn3/blob/23d244df59f5fe313054cfa0c0e4768cd94105fc/elektronn3/training/trainer.py#L281 and https://github.com/ELEKTRONN/elektronn3/blob/23d244df59f5fe313054cfa0c0e4768cd94105fc/elektronn3/data/image.py