Grad-CAM.pytorch icon indicating copy to clipboard operation
Grad-CAM.pytorch copied to clipboard

a little bug in detection/demo.py/gen_cam()

Open liaorongfan opened this issue 4 years ago • 1 comments

def gen_cam(image, mask):
    heatmap = cv2.applyColorMap(np.uint8(255 * mask), cv2.COLORMAP_JET)
    # heatmap = np.float32(heatmap) / 255  # little bug here
    heatmap = np.float32(heatmap)
    heatmap = heatmap[..., ::-1]  # gbr to rgb
    cam = heatmap + np.float32(image)
    return norm_image(cam), heatmap

For variable "heatmap" there is no need to divide by 255.

liaorongfan avatar Dec 03 '20 09:12 liaorongfan

@liaorongfan 感谢您的反馈,image的值范围是0~1,heatmap也缩放到0~1之间,两者相加时在同一数值范围

yizt avatar Dec 09 '20 01:12 yizt