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 Removing Background
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! 🚀