mirror of
https://github.com/BKHMSI/Sketchback.git
synced 2024-07-15 01:50:27 +03:00
22 lines
585 B
Python
Executable File
22 lines
585 B
Python
Executable File
import cv2 as cv
|
|
import numpy as np
|
|
import os
|
|
from filters import PencilSketch
|
|
|
|
m = 240
|
|
n = 320
|
|
DIR_PATH = 'ZuBuD'
|
|
SKETCH_PATH = 'ZuBuD_Sketch'
|
|
|
|
for file in os.listdir(DIR_PATH):
|
|
file_path = os.path.join(DIR_PATH, file)
|
|
img = cv.imread(file_path)
|
|
img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
|
|
height, width, channels = img.shape
|
|
if (height, width, channels) != (m, n, 3):
|
|
img = np.resize(img, (m, n, 3))
|
|
pencil = PencilSketch(width, height)
|
|
sketch = pencil.render(img)
|
|
write_path = os.path.join(SKETCH_PATH, file)
|
|
cv.imwrite(write_path, sketch)
|
|
|