DEV Community

Cover image for Computer Vision is so fun!
Liviu Lupei
Liviu Lupei

Posted on

Computer Vision is so fun!

I work as a Software Engineer at Endtest.

In this article, I will show you how easy it is to use Computer Vision.

The term itself sounds intimidating, it might make you think that you need a PhD in Machine Learning.

Why is it useful?

Computer Vision can be used to detect objects in images.

If you have an Arduino, you can build your own self-driving toy car that detects the road signs.

self-driving toy car

It can also be used to detect differences between images.

This second use case can be extremely useful for developers who want to do automated visual testing.

Let's focus on that one.

Detecting differences between images

Visual testing

The entire concept is dead simple, we just compare each pixel and we calculate the percentage of different pixels.

We’ll just need to make sure our system has Python, OpenCV, scikit-image, Pillow and imutils.

You can find instructions for installing OpenCV here.

As for the rest, you can just use pip:

pip install scikit-image
Enter fullscreen mode Exit fullscreen mode
pip install imutils
Enter fullscreen mode Exit fullscreen mode
pip install pillow
Enter fullscreen mode Exit fullscreen mode

First, we need to make sure that the images have the same size.

The quickest way to achieve that is just to resize the bigger one.

import argparse
import imutils
import cv2
from PIL import Image
from PIL import ImageFilter
from PIL import ImageDraw

image1 = Image.open("/path/to/image1.png")
image2 = Image.open("/path/to/image2.png")

image1_width, image1_height = image1.size
image2_width, image2_height = image2.size

image1_surface = image1_width * image1_height
image2_surface = image2_width * image2_height

if image1_surface != image2_surface:
   if image1_surface > image2_surface:
      image1 = image1.resize((image2_width, image2_height), Image.ANTIALIAS)
      image1.save("/path/to/image1.png")

   if image2_surface > image1_surface:
      image2_surface = image2_surface.resize((image1_width, image1_height), Image.ANTIALIAS)
      image2.save("/path/to/image2.png")
Enter fullscreen mode Exit fullscreen mode

Now our images have the same size.

Screenshot comparison

Time to compare them:

image1 = cv2.imread("/path/to/image1.png")
image2 = cv2.imread("/path/to/image2.png")

i1 = image1
i2 = image2

assert i1.mode == i2.mode, "Different kinds of images."

pairs = izip(i1.getdata(), i2.getdata())
   if len(i1.getbands()) == 1:
      dif = sum(abs(p1 - p2) for p1, p2 in pairs)
   else:
      dif = sum(abs(c1 - c2) for p1, p2 in pairs for c1, c2 in zip(p1, p2))

ncomponents = i1.size[0] * i1.size[1] * 3
diff = (dif / 255.0 * 100) / ncomponents

diff = Decimal(diff)
diff = round(diff, 2)
print "Difference between images is " + str(diff) + "%." 
Enter fullscreen mode Exit fullscreen mode
Difference between images is 7.11%.
Enter fullscreen mode Exit fullscreen mode

We can also generate a third image that shows us the differences.

This can be done by generating a mask and adding it on top of the first image.


difference = cv2.subtract(image1, image2)
Conv_hsv_Gray = cv2.cvtColor(difference, cv2.COLOR_BGR2GRAY)
ret, mask = cv2.threshold(Conv_hsv_Gray, 0, 255,cv2.THRESH_BINARY_INV |cv2.THRESH_OTSU)

difference[mask != 255] = [0, 0, 255]

image1[mask != 255] = [0, 0, 255]
image2[mask != 255] = [0, 0, 255]

cv2.imwrite("/path/to/image3.png", image1)
Enter fullscreen mode Exit fullscreen mode

visual testing

Benefits

There have been documented cases where elements have disappeared from pages and no one noticed it for weeks.

visual testing

Implementing a visual check for your site can't hurt.

screenshot comparison test

And think of how much fun you can have by testing your implementation on all of those Spot the differences challenges.

POC vs Reality

What I showed you was a basic example.

If you would apply this on an actual system, you would also need to make small tweaks.

For example, let's say that your images are identical, but one of them is 2px higher than the other.

You would need a function to focus the 2 images before comparing them.

Almost forgot

If you want to perform Automated Testing without reinventing the wheel, you can use Endtest.

Top comments (0)