如何使用opencv python根据其位置生成检测到的对象的热图

Question

我需要根据其位置生成对象的热图 示例：在视频帧中检测到的绿球。如果它长时间停留在一个位置，那么该位置应该是红色的，并且球在短时间内通过的帧中的位置必须是蓝色的，这样我需要生成一个热图。提前致谢

Answer 1

你可以在这里做的是

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