draw.heat_map

Description

Superimposes a heat map over an image.

class Node(config=None, **kwargs)[source]

Superimposes a heat map over an image.

The draw.heat_map node helps to identify areas that are more crowded. Areas that are more crowded are highlighted in red while areas that are less crowded are highlighted in blue.

Inputs

img (numpy.ndarray): A NumPy array of shape \((height, width, channels)\) containing the image data in BGR format.

density_map (numpy.ndarray): A NumPy array of shape \((H, W)\) representing the number of persons per pixel. \(H\) and \(W\) are the height and width of the input image, respectively. The sum of the array is the estimated total number of people. This is produced by nodes such as model.csrnet.

Outputs

img (numpy.ndarray): A NumPy array of shape \((height, width, channels)\) containing the image data in BGR format.

Configs

None.