utils.py 文件源码

def save_cam_image(img, mask, filename):
    img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
    heatmap = cv2.applyColorMap(np.uint8(255.*mask), cv2.COLORMAP_JET)
    res = np.concatenate((img, heatmap), axis=1)
    cv2.imwrite(filename, res)