In this tutorial, you will learn how to mask images using OpenCV.
My previous guide discussed bitwise operations, a very common set of techniques used heavily in image processing.
And as I hinted previously, we can use both bitwise operations and masks to construct ROIs that are non-rectangular. This allows us to extract regions from images that are of completely arbitrary shape.
Put simply; a mask allows us to focus only on the portions of the image that interests us.
For example, let’s say that we were building a computer vision system to recognize faces. The only part of the image we are interested in finding and describing is the parts of the image that contain faces — we simply don’t care about the rest of the image’s content. Provided that we could find the faces in the image, we may construct a mask to show only the faces in the image.
Another image masking application you’ll encounter is alpha blending and transparency (e.g., in this guide on Creating GIFs with OpenCV). When applying transparency to images with OpenCV, we need to tell OpenCV what parts of the image transparency should be applied to versus not — masks allow us to make that distinction.
To learn how to perform image masking with OpenCV, just keep reading.
Looking for the source code to this post?
Jump Right To The Downloads SectionImage Masking with OpenCV
In the first part of this tutorial, we’ll configure our development environment and review our project structure.
We’ll then implement a Python script to mask images with OpenCV.
Configuring your development environment
To follow this guide, you need to have the OpenCV library installed on your system.
Luckily, OpenCV is pip-installable:
$ pip install opencv-contrib-python
If you need help configuring your development environment for OpenCV, I highly recommend that you read my pip install OpenCV guide — it will have you up and running in a matter of minutes.
Having problems configuring your development environment?
All that said, are you:
- Short on time?
- Learning on your employer’s administratively locked system?
- Wanting to skip the hassle of fighting with the command line, package managers, and virtual environments?
- Ready to run the code right now on your Windows, macOS, or Linux systems?
Then join PyImageSearch University today!
Gain access to Jupyter Notebooks for this tutorial and other PyImageSearch guides that are pre-configured to run on Google Colab’s ecosystem right in your web browser! No installation required.
And best of all, these Jupyter Notebooks will run on Windows, macOS, and Linux!
Project structure
Performing image masking with OpenCV is easier than you think. But before we write any code, let’s first review our project directory structure.
Start by using the “Downloads” section of this guide to access the source code and example image.
Your project folder should look like the following:
$ tree . --dirsfirst . ├── adrian.png └── opencv_masking.py 0 directories, 2 files
Our opencv_masking.py
script will load the input adrian.png
image from disk. We’ll then use masking to extract both the body and face from the image using rectangular and circular masks, respectively.
Implementing image masking with OpenCV
Let’s learn how to apply image masking using OpenCV!
Open the opencv_masking.py
file in your project directory structure, and let’s get to work:
# import the necessary packages import numpy as np import argparse import cv2 # construct the argument parser and parse the arguments ap = argparse.ArgumentParser() ap.add_argument("-i", "--image", type=str, default="adrian.png", help="path to the input image") args = vars(ap.parse_args())
Lines 2-4 import our required Python packages. We then parse our command line arguments on Lines 7-10.
We only need a single switch here, --image
, which is the path to the image we want to mask. We go ahead and default the --image
argument to the adrian.png
file in our project directory.
Let’s now load this image from disk and perform masking:
# load the original input image and display it to our screen image = cv2.imread(args["image"]) cv2.imshow("Original", image) # a mask is the same size as our image, but has only two pixel # values, 0 and 255 -- pixels with a value of 0 (background) are # ignored in the original image while mask pixels with a value of # 255 (foreground) are allowed to be kept mask = np.zeros(image.shape[:2], dtype="uint8") cv2.rectangle(mask, (0, 90), (290, 450), 255, -1) cv2.imshow("Rectangular Mask", mask) # apply our mask -- notice how only the person in the image is # cropped out masked = cv2.bitwise_and(image, image, mask=mask) cv2.imshow("Mask Applied to Image", masked) cv2.waitKey(0)
Lines 13 and 14 load the original image
from disk and display it to our screen:
We then construct a NumPy array, filled with zeros, with the same width and height as our original image on Line 20.
As I mentioned in our previous tutorial on Image cropping with OpenCV, we can use object detection methods to detect objects/people in images automatically. Still, we’ll be using our a priori knowledge of our example image for the time being.
We know that the region we want to extract is in the image’s bottom-left corner. Line 21 draws a white rectangle on our mask, which corresponds to the region we want to extract from our original image.
Remember reviewing the cv2.bitwise_and
function in our bitwise operations tutorial? It turns out that this function is used extensively when applying masks to images.
We apply our mask on Line 26 using the cv2.bitwise_and
function.
The first two parameters are the image
itself (i.e., the image where we want to apply the bitwise operation).
However, the important part of this function is the mask
keyword. When supplied, the bitwise_and
function is True
when the pixel values of the input images are equal, and the mask is non-zero at each (x, y)-coordinate (in this case, only pixels that are part of the white rectangle).
After applying our mask, we display the output on Lines 27 and 28, which you can see in Figure 3:
Using our rectangular mask, we could extract only the region of the image that contains the person and ignore the rest.
Let’s look at another example, but this time using a non-rectangular mask:
# now, let's make a circular mask with a radius of 100 pixels and # apply the mask again mask = np.zeros(image.shape[:2], dtype="uint8") cv2.circle(mask, (145, 200), 100, 255, -1) masked = cv2.bitwise_and(image, image, mask=mask) # show the output images cv2.imshow("Circular Mask", mask) cv2.imshow("Mask Applied to Image", masked) cv2.waitKey(0)
On Line 32, we re-initialize our mask
to be filled with zeros and the same dimensions as our original image.
Then, we draw a white circle on our mask image, starting at the center of my face with a radius of 100
pixels.
Applying the circular mask is then performed on Line 34, again using the cv2.bitwise_and
function.
The results of our circular mask can be seen in Figure 4:
Here, we can see that our circle mask is shown on the left and the application of the mask on the right. Unlike the output from Figure 3, when we extracted a rectangular region, this time, we have extracted a circular region that corresponds to only my face in the image.
Furthermore, we can use this approach to extract regions from an image of arbitrary shape (rectangles, circles, lines, polygons, etc.).
OpenCV image masking results
To perform image masking with OpenCV, be sure to access the “Downloads” section of this tutorial to retrieve the source code and example image.
From there, open a shell and execute the following command:
$ python opencv_masking.py
Your masking output should match mine from the previous section.
OpenCV Image Masking is powerful for manipulating images. It allows you to apply effects to a single image and create an entirely new look.
With OpenCV Image Masking, you can selectively modify colors, contrast, lighten or darken, add or remove noise, and even erase parts or objects from an image. It is also possible to add text, and special effects, and even convert images to a different file format.
OpenCV Image Masking is a great way to easily create stunning visuals and might help you with:
- graphic design
- application development
- robotics
- autonomy
- computer vision research topics
What's next? We recommend PyImageSearch University.
86 total classes • 115+ hours of on-demand code walkthrough videos • Last updated: October 2024
★★★★★ 4.84 (128 Ratings) • 16,000+ Students Enrolled
I strongly believe that if you had the right teacher you could master computer vision and deep learning.
Do you think learning computer vision and deep learning has to be time-consuming, overwhelming, and complicated? Or has to involve complex mathematics and equations? Or requires a degree in computer science?
That’s not the case.
All you need to master computer vision and deep learning is for someone to explain things to you in simple, intuitive terms. And that’s exactly what I do. My mission is to change education and how complex Artificial Intelligence topics are taught.
If you're serious about learning computer vision, your next stop should be PyImageSearch University, the most comprehensive computer vision, deep learning, and OpenCV course online today. Here you’ll learn how to successfully and confidently apply computer vision to your work, research, and projects. Join me in computer vision mastery.
Inside PyImageSearch University you'll find:
- ✓ 86 courses on essential computer vision, deep learning, and OpenCV topics
- ✓ 86 Certificates of Completion
- ✓ 115+ hours of on-demand video
- ✓ Brand new courses released regularly, ensuring you can keep up with state-of-the-art techniques
- ✓ Pre-configured Jupyter Notebooks in Google Colab
- ✓ Run all code examples in your web browser — works on Windows, macOS, and Linux (no dev environment configuration required!)
- ✓ Access to centralized code repos for all 540+ tutorials on PyImageSearch
- ✓ Easy one-click downloads for code, datasets, pre-trained models, etc.
- ✓ Access on mobile, laptop, desktop, etc.
Summary
You came to learn Computer Vision and basic masking, super! In this tutorial, you learned the basics of masking using OpenCV.
The key point of masks is that they allow us to focus our computation only on regions of the image that interest us. Focusing our computations on regions that interest us dramatically impacts when we explore topics such as machine learning, image classification, and object detection.
For example, let’s assume that we wanted to build a system to classify the species of the flower.
In reality, we are probably only interested in the flower petals’ color and texture to perform the classification. But since we are capturing the photo in a natural environment, we’ll also have many other regions in our image, including dirt from the ground, insects, and other flowers crowding the view. How will we quantify and classify just the flower we are interested in? As we’ll see, the answer is masks.
Updated: December 30th, 2022 with updated links and content.
To download the source code to this post (and be notified when future tutorials are published here on PyImageSearch), simply enter your email address in the form below!
Download the Source Code and FREE 17-page Resource Guide
Enter your email address below to get a .zip of the code and a FREE 17-page Resource Guide on Computer Vision, OpenCV, and Deep Learning. Inside you'll find my hand-picked tutorials, books, courses, and libraries to help you master CV and DL!
Comment section
Hey, Adrian Rosebrock here, author and creator of PyImageSearch. While I love hearing from readers, a couple years ago I made the tough decision to no longer offer 1:1 help over blog post comments.
At the time I was receiving 200+ emails per day and another 100+ blog post comments. I simply did not have the time to moderate and respond to them all, and the sheer volume of requests was taking a toll on me.
Instead, my goal is to do the most good for the computer vision, deep learning, and OpenCV community at large by focusing my time on authoring high-quality blog posts, tutorials, and books/courses.
If you need help learning computer vision and deep learning, I suggest you refer to my full catalog of books and courses — they have helped tens of thousands of developers, students, and researchers just like yourself learn Computer Vision, Deep Learning, and OpenCV.
Click here to browse my full catalog.