熱力圖展示:
myplot.png
多元正太分布生成2d的heatmap送火， 多個中心點生成的heatmap 進行累積拳话，最后輸出多個中心點的熱力圖。
# coding=utf-8
import numpy as np
import matplotlib.pyplot as plt
import random
from scipy.stats import multivariate_normal
covs =[40,50,60] # 可選cov list
def CenterLabelHeatMap(img_width, img_height, c_x, c_y, sigma):
X1 = np.linspace(1, img_width, img_width)
Y1 = np.linspace(1, img_height, img_height)
[X, Y] = np.meshgrid(X1, Y1)
X = X - c_x
Y = Y - c_y
D2 = X * X + Y * Y
E2 = 2.0 * sigma * sigma
Exponent = D2 / E2
heatmap = np.exp(-Exponent)
return heatmap
def heatmap_n_point(points, size, scaler=30):
"""
根據(jù)中心點生成熱力值（基于multivariate_normal）
:param points: 2d list [[point1], point2]
:param size: tuple (h,w)
:param scaler: 熱力值縮放因子种吸，默認為1(調(diào)整方差大小)
:return: np.array
"""
assert isinstance(size, tuple), "size 輸入錯誤"
assert isinstance(points, list) and isinstance(points[0], list), "points input error"
xx, yy = np.meshgrid(range(size[1]), range(size[0]))
# evaluate kernels at grid points
xxyy = np.c_[xx.ravel(), yy.ravel()]
kernel = 0.0
for point in points:
kernel += multivariate_normal(point, scaler*random.choice([90,80,100])).pdf(xxyy)
# kernel += CenterLabelHeatMap(size[0], size[1], point[0], point[1],40)
return kernel.reshape(size)
if __name__ == '__main__':
import time
s = time.time()
img = heatmap_n_point([[40,30], [400,70], [200, 800], [80,300]], (800, 1200))
print(time.time()-s)
plt.imshow(img)
plt.show()
