根据关键点坐标生成heatmap

使用高斯模糊，根据关键点坐标生成heatmap
假设heatmap大小为(64,48)，关键点的坐标为(32,24)
即将一个(64,48)的黑图的中心点点亮

import matplotlib.pyplot as plt
import numpy as np
import cv2

def generate_heatmap(heatmap, sigma):

    heatmap[32][24] = 1
    heatmap = cv2.GaussianBlur(heatmap, sigma, 0)
    am = np.amax(heatmap)
    heatmap /= am / 255
    return heatmap


target = np.zeros((64, 48))
plt.imshow(target, cmap='hot', interpolation='nearest')
plt.show()
#target = generate_heatmap(target, (15,15))
#target = generate_heatmap(target, (11,11))
#target = generate_heatmap(target, (9,9))
target = generate_heatmap(target, (7,7))
plt.imshow(target, cmap='hot', interpolation='nearest')
plt.show()
print(target)

sigma为(15,15)

sigma为（11，11）

sigma为（9，9）

sigma为（7，7）