SIGGRAPH2022/DCT-Net

1 2import cv23from source.cartoonize import Cartoonizer4import os5 6def process():7 8    algo = Cartoonizer(dataroot='damo/cv_unet_person-image-cartoon_compound-models')9    img = cv2.imread('input.png')[...,::-1]10 11    result = algo.cartoonize(img)12 13    cv2.imwrite('res.png', result)14    print('finished!')15 16 17 18 19if __name__ == '__main__':20    process()21 22 23 24