使用高斯模糊,根据关键点坐标生成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)