search

hans66hsu / nn_interpretability

Pytorch implementation of various neural network interpretability methods
MIT License
110 stars 20 forks source link

LRP for semantic segmentation #1

Closed bugsuse closed 1 year ago

bugsuse commented 3 years ago

Nice work! I want to apply LRP to unet model for explaining semantic segmentation tasks. I looked up most implementations of the Pytorch version of LRP and almost none of them support semantic segmentation tasks. Does the LRP in this resource support the interpretation of semantic segmentation tasks? Thanks a lot!

hans66hsu commented 3 years ago

The current focus of the examples is classification tasks. To extend it to segmentation tasks you would have to change the network to a fully convolutional network but the LRP rule in our package should be applicable.