Convert Image Files To Line Drawings With OpenCV

convert_img_to_line_drawing.py

# -*- coding: utf-8 -*-

import os

import cv2
import numpy as np

def create_line_drawing_image(img):
    kernel = np.array([
        [1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1],
        ], np.uint8)
    img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    img_dilated = cv2.dilate(img_gray, kernel, iterations=1)
    img_diff = cv2.absdiff(img_dilated, img_gray)
    contour = 255 - img_diff
    return contour

def convert_images(dir_from, dir_to):
    for file_name in os.listdir(dir_from):
        if file_name.endswith('.JPG'):
            print(file_name)
            img = cv2.imread(os.path.join(dir_from, file_name))
            img_contour = create_line_drawing_image(img)
            cv2.imwrite(os.path.join(dir_to, file_name), img_contour)

if __name__ == '__main__':
    dir_src = '/path/to/source/directory'
    dir_dest = '/path/to/dest/directory'
    convert_images(dir_src, dir_dest)