Introduction
Hello! In this tutorial I will show how to apply a simply cartoon filter on any image 😃
Preparing the environment
First we need to initiate the python virtual environment which can be done via:
python3 -m venv env
source env/bin/activate
Installing the dependencies
Create a requirements.txt file and add the following:
# requirements.txt
opencv-python
Then install via:
pip install -r requirements.txt
Writing the source code
Importing the neccessary modules
import argparse
import cv2
Writing the function to apply the cartoon effect
def cartoonize (image):
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
blurImage = cv2.medianBlur(image, 1)
edges = cv2.adaptiveThreshold(gray, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 9, 9)
color = cv2.bilateralFilter(image, 9, 200, 200)
cartoon = cv2.bitwise_and(color, color, mask = edges)
return cartoon
This function basically takes an image and applies a cartoonish style filter to the image which is then returned.
Median blue is most effective when it comes to reducing the noise from the image.
Adaptive Thresholding is used to overcome false predictions and illumination problems with the image.
Bilateral filter is a edge preserving and smoothing filter for images.
Finally we apply the filter onto the image via bitwise_and.
Finally we need to write the main function
if __name__ == "__main__":
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--image", required = True, help = "Path to input file")
args = vars(ap.parse_args())
image = cv2.imread(args["image"])
cartoonImage = cartoonize(image)
cv2.imwrite("output.jpg", cartoonImage)
cv2.imshow("output", cartoonImage)
cv2.waitKey(0)
cv2.destroyAllWindows()
What we do here is basically parse the image argument from the command line, apply the cartoon filter and then write the output to file and then show the output to the user.
After that we listen for any key and then cleanup the window.
Running the program
The program can be run via:
python main.py -i original.jpg
If all goes well you should see the following output:
The sample image was taken from my home town in the UK. 😎
Conclusion
Here I have demonstrated how to apply a simple cartoon like filter to an image, feel free to experiment with your own images are even expand it to include more filters. 😀
Happy Coding!