search

Shreeyak / pytorch-lightning-segmentation-template

Semantic Segmentation on the LaPa dataset using Pytorch Lightning
Apache License 2.0
13 stars 5 forks source link

Log images in callback #33

Closed Shreeyak closed 3 years ago

Shreeyak commented 3 years ago

Use a pl callback to log segmentation results (images) to wandb. Screenshot from 2020-12-29 14-27-49

Shreeyak commented 3 years ago

Fixes https://github.com/Shreeyak/pytorch-lightning-segmentation-template/pull/33 and https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/43

Shreeyak commented 3 years ago

Fixes #44