| import cv2 | |
| from modelscope.outputs import OutputKeys | |
| from modelscope.pipelines import pipeline | |
| from modelscope.utils.constant import Tasks | |
| img_colorization = pipeline(Tasks.image_colorization, model='damo/cv_ddcolor_image-colorization') | |
| result = img_colorization('https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/audrey_hepburn.jpg') | |
| cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG]) |