How to Remove Background from Image using Python

In this tutorial, you will learn how to remove background from an image using Python. We will use the rembg library to remove the background from an image and make a PNG image with a transparent background.

We will use the rembg and Pillow libraries to remove the background from an image.

Install rembg library

pip install rembg

The rembg library uses the u2net model to remove the background from an image. The u2net model is a deep learning model that is trained to remove the background from an image.

This library directly depends on the onnxruntime library. Therefore, we can only update the Python version when onnxruntime provides support for that specific version.

Install Pillow library

pip install pillow

Remove Background from Image using Python

from rembg import remove
from PIL import Image

image_path = 'cat.jpg'

input = Image.open(image_path)

output = remove(input)

output.save('cat_transparent.png', 'PNG')

After removing the background from the image, the output image will be saved with a transparent background.

before remove bg image Before Removing Background

after remove bg image After Removing Background

In the above code, we first import the remove function from the rembg library and the Image class from the Pillow library.

We then open the image using the Image.open method and pass the image path to it.

We then call the remove function and pass the input image to it. This function will remove the background from the image and return the image with a transparent background.

We then save the output image using the save method and pass the output image path and the image format to it.

Now, you can run the above code and it will remove the background from the image and save the output image with a transparent background.

That’s it. You have successfully removed the background from an image using Python.

I hope this tutorial helps you to remove the background from an image using Python. If you have any questions, please feel free to ask in the comments section below.

Happy Coding! 🚀