关于图6(b)-(d)绘制方法的疑问

[huajichun]

您好，感谢您出色的工作！
我想请教一下关于图6(b)-(d)的绘制方法。
感谢您的帮助，期待您的回复！

[Linwei-Chen]

您好，感谢您对我们工作的关注！

关于图 6(b)-(d) 的绘制方法，主要是对论文中提出的 FBM (Frequency Band Modulation) 模块中生成的空间调制图 (Spatial Modulation Maps) 进行可视化。

具体步骤如下：

1. 数据获取：
   在代码的 FBM 模块（Frequency Band Modulation）的 forward 函数中，找到生成调制系数的分支。这个分支通常经过一个卷积层后接 Sigmoid 激活函数，对应论文公式 (3) 中的 $\mathbf{A}_b$。您需要提取这个经过 Sigmoid 之后的 Tensor（数值在 0 到 1 之间）。

2. 可视化绘制：
   获取到 Tensor 后（假设形状为 [H, W]），可以使用 Python 的 matplotlib 库进行热力图绘制。

   • 将 Tensor 转为 numpy 数组。
   • 使用 plt.imshow() 函数，并设置 colormap（论文中看起来使用的是 'jet' 或类似的高对比度色阶）。
   • 如果需要叠加在原图上，可以将热力图调整大小后以一定的透明度（alpha blend）叠加在输入图像上。

简单代码示例：

import matplotlib.pyplot as plt
import cv2

# 假设 modulation_map 是你在 FBM 模块中提取出的某一个频段的调制图，shape 为 (H, W)
# modulation_map = ... (从模型 forward 中 hook 出来)

plt.figure()
# 使用 jet 色阶来模拟论文中的效果
plt.imshow(modulation_map, cmap='jet') 
plt.colorbar()
plt.axis('off')
plt.show()

希望能对您有所帮助！