如何使用python和openCV制作热图

时间:2016-11-05 09:41:48

标签: python opencv matplotlib heatmap

我正在努力实现这个:

YouTube链接 - https://www.youtube.com/watch?v=Qk4V_x6B7jY&t=5s

博客链接 - http://www.businessinsider.in/These-Heat-Maps-Show-How-Retailers-Track-You-As-You-Shop/articleshow/29512380.cms

我想使用python和openCV,但我是openCV的初学者,因此,我不知道如何实现它。

我有一些基本的想法。我已经能够跟踪运动并在移动对象周围绘制一个矩形,我将矩形的坐标保存在外部csv文件中。但是,我被困在绘制热图部分。如何制作它,以便随着时间的推移人们在一个区域内移动越来越多,颜色从蓝色(正常运动)变为红色(高运动)?请帮忙..

1 个答案:

答案 0 :(得分:2)

因此,绘制热图的基本思路是在视觉上获得对给定特定事件概率的一些反馈,您可以编写自己的方法,该方法可能在0-1范围内取概率并输出范围内的颜色(255 ,0,0) - (0,0,255)。或Opencv提供color-maps。您可能有兴趣使用 COLORMAP_JET

enter image description here

现在你必须将范围0-255而不是0-1的概率标准化,你可以使用cv2.applyColorMap(input_prob, cv2.COLORMAP_JET)来获得所需的输出。