OpenCV 2
In [ ]:
#!pip install opencv-python-headless numpy
In [1]:
import cv2
import imageio.v2 as imageio
import numpy as np
import matplotlib.pyplot as plt
def plot_image_from_array(image_array, title='Image'):
plt.imshow(image_array, cmap='gray')
plt.axis('off') # Hide axes
plt.title(title)
plt.show()
In [2]:
def dodgeV2(image, mask):
return cv2.divide(image, 255 - mask, scale=256)
In [3]:
def pencil_sketch(image_path):
img = cv2.imread(image_path, cv2.IMREAD_COLOR)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
inverted_gray = 255 - gray
blurred = cv2.GaussianBlur(inverted_gray, (21, 21), sigmaX=0, sigmaY=0)
sketch = dodgeV2(gray, blurred)
return sketch
plot_image_from_array(pencil_sketch('Sharbat_Gula.jpg'))
In [4]:
def convert_to_line_drawing(image_path):
img = cv2.imread(image_path)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
blurred = cv2.GaussianBlur(gray, (5, 5), 0)
edges = cv2.Canny(blurred, 50, 150)
inverted_edges = cv2.bitwise_not(edges)
return inverted_edges
plot_image_from_array(convert_to_line_drawing('Sharbat_Gula.jpg'))
