Spaces:
Runtime error
Runtime error
| 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() | |