mirror of https://github.com/menyifang/DCT-Net
add input sample
parent
37112dfce8
commit
2416932f44
@ -1,12 +1,14 @@
|
|||||||
import cv2
|
import cv2
|
||||||
|
from modelscope.hub.snapshot_download import snapshot_download
|
||||||
from modelscope.pipelines import pipeline
|
from modelscope.pipelines import pipeline
|
||||||
from modelscope.utils.constant import Tasks
|
|
||||||
|
|
||||||
img_cartoon = pipeline(Tasks.image_portrait_stylization, 'damo/cv_unet_person-image-cartoon_compound-models')
|
model_dir = snapshot_download('damo/cv_unet_person-image-cartoon_compound-models', cache_dir='.')
|
||||||
img_cartoon = pipeline('image-portrait-stylization')
|
img_cartoon = pipeline('image-portrait-stylization', model=model_dir)
|
||||||
|
|
||||||
result = img_cartoon('input.png')
|
result = img_cartoon('input.png')
|
||||||
|
|
||||||
cv2.imwrite('result.png', result['output_img'])
|
cv2.imwrite('result.png', result['output_img'])
|
||||||
|
print('finished!')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue