DCT-Net/run.py

24 lines
352 B
Python

import cv2
from source.cartoonize import Cartoonizer
import os
def process():
algo = Cartoonizer(dataroot='damo/cv_unet_person-image-cartoon_compound-models')
img = cv2.imread('input.png')[...,::-1]
result = algo.cartoonize(img)
cv2.imwrite('res.png', result)
print('finished!')
if __name__ == '__main__':
process()