https://www.modelscope.cn/models/iic/cv_ddcolor_image-colorization/summary#ddcolor-%E5%9B%BE%E5%83%8F%E4%B8%8A%E8%89%B2%E6%A8%A1%E5%9E%8B

下载模型

pip install modelscope


modelscope download --model 'iic/cv_ddcolor_image-colorization' --local_dir './DDColormodel'

推理

pytroch环境：

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y

pip install addict modelscope datasets sympy==1.13.1 simplejson sortedcontainers timm

推理：

python
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='./DDColormodel')
img_path = 'https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/audrey_hepburn.jpg'
result = img_colorization(img_path)
cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG])

原图：

docker环境

docker run --gpus all --shm-size=32g -it --net host -v ./:/ddcolor kevinchina/deeplearning:2.5.1-cuda12.1-cudnn9-devel-pix2pix bash

pip install addict modelscope datasets sympy==1.13.1 simplejson sortedcontainers timm opencv-python

apt-get update
apt-get install -y libgl1-mesa-glx libjpeg-dev libpng-dev libtiff-dev libopencv-dev

docker commit 731cbf160933 kevinchina/deeplearning:2.5.1-cuda12.1-cudnn9-devel-ddcolor

docker push kevinchina/deeplearning:2.5.1-cuda12.1-cudnn9-devel-ddcolor