gradcam for binary segmentaion network

jacobgil / pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

https://jacobgil.github.io/pytorch-gradcam-book

MIT License

9.79k stars 1.52k forks source link

gradcam for binary segmentaion network #476

Open Nastu-Ho opened 5 months ago

Nastu-Ho commented 5 months ago

The final output of my segmentation network is a single-channel prediction map, how can this situation be adjusted according to the tutorial?