小编bro*_*rum的帖子

从热图数据生成边界框

我有我正在从事的车辆检测项目的热图数据,但我不知道下一步该去哪里。我想在图像的“最热”部分周围绘制一个通用边界框。我的第一个想法是在所有重叠的部分上画一个框,但有些东西告诉我有一种更准确的方法来做到这一点。任何帮助,将不胜感激!不幸的是,我的声誉阻止我发布图像。这是我创建热图的方法:

# Positive prediction window coordinate structure: ((x1, y1), (x2, y2))
def create_heatmap(bounding_boxes_list):
     # Create a black image the same size as the input data
     heatmap = np.zeros(shape=(375, 1242))
     # Traverse the list of bounding box locations in test image
     for bounding_box in bounding_boxes_list:
          heatmap[bounding_box[0][1]:bounding_box[1][1], bounding_box[0][ 0]:bounding_box[1][0]] += 1

return heatmap
Run Code Online (Sandbox Code Playgroud)

这是我拥有的热图的链接

这是我的总体想法

python opencv numpy object-detection

6
推荐指数
1
解决办法
5105
查看次数

标签 统计

numpy ×1

object-detection ×1

opencv ×1

python ×1