如何将特征图的可视化为灰色

特征图的可视化，显示的风格不同

我的解答思路和尝试过的方法

采取hook的方式获取网络的特征，进行特征的可视化，结果如下

想要达到的结果

理想的效果如下，期望是灰色的背景，而不是黑色：

我应该如何获得期望风格的可视化结果？