DEV Community

Kuba
Kuba

Posted on β€’ Edited on

5 1

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.

Billboard image

The fastest way to detect downtimes

Join Vercel, CrowdStrike, and thousands of other teams that trust Checkly to streamline monitoring.

Get started now

Top comments (1)

Collapse
 
kristinides profile image
Kristin Ides DeMar β€’

Nice!

A Workflow Copilot. Tailored to You.

Pieces.app image

Our desktop app, with its intelligent copilot, streamlines coding by generating snippets, extracting code from screenshots, and accelerating problem-solving.

Read the docs

πŸ‘‹ Kindness is contagious

Dive into an ocean of knowledge with this thought-provoking post, revered deeply within the supportive DEV Community. Developers of all levels are welcome to join and enhance our collective intelligence.

Saying a simple "thank you" can brighten someone's day. Share your gratitude in the comments below!

On DEV, sharing ideas eases our path and fortifies our community connections. Found this helpful? Sending a quick thanks to the author can be profoundly valued.

Okay