使用点积的热图可视化

[55renchen]

您好，我用点积版本的non_local对f进行softmax之后，
f = torch.matmul(theta_x, phi_x)
N = f.size(-1)
f_div_C = f / N
sft = nn.Softmax(dim=2)
f_div_C = sft(f_div_C)
在可视化时还需要对NL_MAP进行normalize吗？如果需要，应该如何normalize呢？
现在我的热图结果只有一个box，看不到正常热图颜色的变化，这是什么原因呢？

[AlexHex7]

Hi @55renchen , NL_MAP需要归一化到[0, 1]. uint8类型范围是[0, 255]，超出该范围则溢出

Non-local_pytorch/nl_map_vis/nl_map_vis.py

Line 35 in 1fdae52

nl_map = np.uint8(nl_map * 255)