Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Convert Image Files To Line Drawings With OpenCV
# -*- 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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment