Implementation for OCTAve: 2D en face Optical Coherence Tomography Angiography Vessel Segmentation in Weakly-Supervised Learning with Locality Augmentation (IEEE Transactions on Biomedical Engineering))
Need to add a logging method that supports image logging in training with Lightning.
Discussion
Logging is needed. Normally we can just use Lightning's built-in dashboard as usual. However, for Image logging for attention visualization. We need to convert the tensor into a PIL image. In which, I'm not sure about performance impact.
I'm currently considering the usage of visdom logger. But I'm not sure we can easily integrate it with lightning though.
Proposal
Need to add a logging method that supports image logging in training with Lightning.
Discussion
Logging is needed. Normally we can just use Lightning's built-in dashboard as usual. However, for Image logging for attention visualization. We need to convert the tensor into a PIL image. In which, I'm not sure about performance impact.
I'm currently considering the usage of visdom logger. But I'm not sure we can easily integrate it with lightning though.