人识别及场景热力图绘制

1、首先识别图片中的人，得到bounding box的中心坐标。当然，这个坐标也可以是上中部等都OK，自己定义。
2、将所有中心坐标放入一个list类型的变量data中，即data = [[x1,y1] [x2,y2] …]
3、绘制热力图，并将热力图加权叠加到原图上。
需要的包：

import cv2
import numpy as np
from PIL import Image, ImageFont, ImageDraw
from pyheatmap.heatmap import HeatMap
import matplotlib.pyplot as plt

def apply_heatmap(image,data):
	'''image是原图，data是坐标'''
	'''创建一个新的与原图大小一致的图像，color为0背景为黑色。这里这样做是因为在绘制热力图的时候如果不选择背景图，画出来的图与原图大小不一致（根据点的坐标来的），导致无法对热力图和原图进行加权叠加，因此，这里我新建了一张背景图。'''
    background = Image.new("RGB", (image.shape[1], image.shape[0]), color=0)
    # 开始绘制热度图
    hm = HeatMap(data)
    hit_img = hm.heatmap(base=background, r = 100) # background为背景图片，r是半径，默认为10
    # ~ plt.figure()
    # ~ plt.imshow(hit_img)
    # ~ plt.show()
    #hit_img.save