我需要根据一个对象的位置生成它的热图示例:视频帧中检测到的绿色球。如果它在一个位置停留了很长一段时间,那么这个位置应该是红色的,而在帧中球经过的时间短的位置必须是蓝色的,这样我就需要生成一个热图。提前感谢
你可以在这里做的是
1)首先将热图定义为图像的大小
heatmap = np.zeros_like(img[:,:,0]).astype(np.float)
2)因为你已经有了被检测的对象和它的位置,在热图中,在对象的边界框中添加1的所有像素。
heatmap[box[0][1]:box[1][1], box[0][0]:box[1][0]] += 1
3)你可以通过设置热图中低于阈值的所有像素为0来应用阈值。您可以选择阈值为1,这样边界框内的区域将保留在热图
中。heatmap[heatmap <= threshold] = 0