Basic Image filter using OpenCV
import cv2
class Cartoonizer():
def __init__(self):
pass
def render(self,img_rgb):
img_rgb = cv2.imread(img_rgb)
#img_rgb = cv2.resize(img_rgb, (1366,768))
numDownSamples = 2
numBilateralFilters = 50
for _ in range(numDownSamples):
img_color = cv2.pyrDown(img_rgb)
for _ in range(numBilateralFilters):
img_color = cv2.bilateralFilter(img_color ,9,9,7)
for _ in range(numDownSamples):
img_color = cv2.pyrUp(img_color)
img_gray = cv2.cvtColor(img_rgb , cv2.COLOR_RGB2GRAY)
img_blur = cv2.medianBlur(img_gray , 3)
img_edge = cv2.adaptiveThreshold(img_blur ,255,
cv2.ADAPTIVE_THRESH_MEAN_C,
cv2.THRESH_BINARY , 9 ,2)
(x ,y ,z) = img_color.shape
img_edge = cv2.resize(img_edge , (y,x))
img_edge = cv2.cvtColor(img_edge , cv2.COLOR_GRAY2RGB)
cv2.imwrite("edge.png" , img_edge)
return cv2.bitwise_and(img_color , img_edge)
tmp_canvas = Cartoonizer()
file_name = "me.png"
res = tmp_canvas.render(file_name)
cv2.imwrite("Cartoon version.jpg",res)
cv2.imshow("Cartoon version",res)
cv2.waitKey(0)
cv2.destroyAllWindows()
Comments
Post a Comment