DEV Community

Kuba
Kuba

Posted on • Updated on

Make yourself a heatmap πŸ•β€πŸ¦Ί

Data

Obviously if we talk about heatmaps, we talk about displaying some data. Today it is about displaying heatmap on an image. In this case it will be dog position in the garden. You can read how to obtain such position here. Connect the dots, and you will have some cool system in your garden 😎.

First, I have to admit that I don't have a dog. Data will be generated randomly X and Y positions.

import numpy as np

# 100 random pairs of x and y in range from 0 to 20
data = np.random.randint(0,20,(100,2))
Enter fullscreen mode Exit fullscreen mode

Image to put heatmap on it

Now I will use my talent a bit 😎. Hope you can see a garden here.

garden

Load image and put data strait forward on it

Let's use OpenCV and add some dots. Assume that our garden is 20 m long. Scale coordinates to shape of our image.

import numpy as np 
import cv2 

data = np.random.randint(0,20,(100,2))

map_img = cv2.imread("HERE PUT PATH TO IMG ON DISC")
width, height, _ = map_img.shape 

for coord in data:
    x, y = coord
    x = int(( x / 20 ) * width)
    y = int(( y / 20 ) * height)
    cv2.circle( map_img, (x,y), 20, (255,0,0), -1  )

while True:
    cv2.imshow("map", map_img )
    if cv2.waitKey(1) & 0xFF == ord("q"):
        break

cv2.destroyAllWindows()
Enter fullscreen mode Exit fullscreen mode

garden

Hmm. There is some data, but it's hard to analyze it. Let's dig deeper.

Let's make second image, blank one. We will put our data on it, manipulate a bit and then overlay on target image.

Blank image is quite easy in OpenCV. Just matrix with zeros.

heatmap_image = np.zeros((height,width,1), np.uint8) 
Enter fullscreen mode Exit fullscreen mode

Putting data on it is the same as above. Use


. You should be able to display such dots.

![garden](https://github.com/JakubSzwajka/JakubSzwajka.github.io/blob/master/_posts/_images/garden_3.png?raw=true)

### Manipulate them

Use ``cv2.distanceTransform()`` for all pixels. It changes pixels value based on distance to the nearest pixel with value 0. So if there is a lot of points in one place, value will be higher.




```python
heatmap_image = cv2.distanceTransform(heatmap_image, cv2.DIST_L2, 5)
Enter fullscreen mode Exit fullscreen mode

garden

Quite good! Now let's add some color. To make in more readable, in this example I'm multiplying every pixel by 2.5. Change this value and see what happens. πŸ˜‰

# here I make those points a bit bigger
heatmap_image = heatmap_image * 2.5
heatmap_image = np.uint8(heatmap_image)
heatmap_image = cv2.applyColorMap(heatmap_image, cv2.COLORMAP_JET)
Enter fullscreen mode Exit fullscreen mode

Remember that cv2.distanceTransform( ) change type of data a bit. We have to change it back to np.uint8.

Overlay

Final step. Overlay those two images.

fin_img = cv2.addWeighted(heatmap_image, 0.5, map_img, 0.5, 0)
Enter fullscreen mode Exit fullscreen mode

Garden

We can see the data now! We can assume where our dog spends most of the time. πŸ‘

Try to connect this with this post, and you can make quite interesting camera system πŸ€”. You can try changing values in distanceTransform( ) too. For example try different distance types. They will change your heatmap a bit.

Discussion (1)

Collapse
kristinides profile image
kristin ides

Nice!