[Picture to sketch] Small python code to draw a sketch from a picture #picture

picture_to_sketch.py

import cv2

img_location = './'
filename = 'profil.jpg'

img = cv2.imread(img_location + filename)

def picture_to_sketch(img, color=False ,blur_kernel=40):
    if not color :
        img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) 

    inverted_picture = cv2.bitwise_not(img)
    blur_picture = cv2.blur(inverted_picture, (blur_kernel, blur_kernel))
    sketch = cv2.divide(img, 255 - blur_picture,  scale=256)
    return sketch

sketch = picture_to_sketch(img)
sketch_color = picture_to_sketch(img, color=True)

cv2.imwrite('sketch.jpg', sketch)
cv2.imwrite('sketch_color.jpg', sketch_color)