def save_heatmap(output_name):
    (output, name) = output_name
    fig = plt.figure()
    plt.axis('off')
    heatmap = plt.imshow(output, cmap='viridis')
    fig.colorbar(heatmap)
    fig.savefig(('%s_heatmap.png' % name.split('.jpg')[0]))
    return
def save_heatmap(output_name):
    (output, name) = output_name
    fig = plt.figure()
    plt.axis('off')
    heatmap = plt.imshow(output, cmap='viridis')
    fig.colorbar(heatmap)
    fig.savefig(('%s_heatmap.png' % name.split('.jpg')[0]))
    return