如何在多任务网络中使用 gradcam

作者: regedit818创建于 2025年8月31日更新于 2025年8月31日

你好,我的网络采用单个编码器和双编码器结构,一个编码器输出掩码,另一个编码器输出边缘。我重载了 GradCam 的前向函数。我也尝试了各种组合,例如边缘损失 + 掩码损失,或者仅使用掩码损失。然而,可视化结果并不令人满意(具体来说,会出现全局激活的现象)。同时,我发现单个编码器网络的可视化情况正常。

内容来源: jacobgil/pytorch-grad-cam