mirror of https://github.com/menyifang/DCT-Net
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
12 lines
391 B
Python
12 lines
391 B
Python
import cv2
|
|
from modelscope.outputs import OutputKeys
|
|
from modelscope.pipelines import pipeline
|
|
from modelscope.utils.constant import Tasks
|
|
|
|
img_cartoon = pipeline(Tasks.image_portrait_stylization,
|
|
model='damo/cv_unet_person-image-cartoon_compound-models')
|
|
result = img_cartoon('input.png')
|
|
|
|
cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG])
|
|
print('finished!')
|