还在担心黑白老照片没有颜色丨基于ModelArts实例感知图像上色丨【华为云AI贺新年】

前言

ModelArts 是面向开发者的一站式 AI 开发平台，为机器学习与深度学习提供海量数据预处理及交互式智能标注、大规模分布式训练、自动化模型生成，及端-边-云模型按需部署能力，帮助用户快速创建和部署模型，管理全周期 AI 工作流。

前期准备

在使用ModelArts之前，需要进入华为云官网 https://www.huaweicloud.com/ ，然后注册华为云账号，再进行实名认证。主要分为4步（注册–>实名认证–>服务授权–>领代金券）（如有已完成部分，请忽略）

点此去完成 实名认证，账号类型选"个人"，个人认证类型推荐使用"扫码认证"。

具体步骤

下面是给黑白图片上色的步骤。

第一步，点击下方连接进入 实例感知图像上色 的 JupyterLab 页面。

https://authoring-modelarts-cnnorth4.huaweicloud.com/console/lab?share-url-b64=aHR0cHM6Ly9jbm5vcnRoNC1tb2RlbGh1Yi1tb2RlbHMub2JzLmNuLW5vcnRoLTQubXlod2Nsb3Vkcy5jb20vaXRlbS83OTJjNzlhNS1jYzI5LTRhNjUtYmUzNy01MTRhYjk1OWE2OTIvMS4wLjAvSW5zdENvbG9yaXphdGlvbi5pcHluYg%3D%3D&galleryitemid=792c79a5-cc29-4a65-be37-514ab959a692

之后配置当前运行环境，进行 规格切换，并选择 [限时免费]GPU: 1*V100|CPU: 8核 64GB ，点击切换规格。

切换中，等待切换成功。

资源切换完成，点击确定。

第二步，下载代码和数据

选中以下代码，Ctrl+Enter 运行

也可以使用上方工具栏中的三角形按钮。

运行成功后的结果：

第三步，安装依赖库

按照第二步的方法，选中以下代码运行。

等待一段时间，当发现左下方的 Busy 变成 Idle 时，表示运行结束。

第四步，添加本地jpg格式的黑白图片

注意是在 example 文件夹下添加，我这里添加的是一张蒲公英的照片（dandelion.jpg）

第五步，上色

1、转到 InstColorization 目录下。

cd InstColorization/

2、配置 Detectron2。

from os.path import join, isfile, isdir
from os import listdir
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
from argparse import ArgumentParser

import detectron2
from detectron2.utils.logger import setup_logger
setup_logger()

import numpy as np
import cv2

# import some common detectron2 utilities
from detectron2 import model_zoo
from detectron2.engine import DefaultPredictor
from detectron2.config import get_cfg

import torch

cfg = get_cfg()
cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml"))
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7
cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml")
predictor = DefaultPredictor(cfg)

这一步需要等待一段时间。

成功之后，运行以下三段代码。当出现 dandelion.npz 说明本地图片已经配置完毕。（dandelion是我上传的本地图片的名字）

开始进行图像上色，继续运行以下几段代码。

第六步，展示上色效果

需要修改的地方：

img_name_list = ['000000022969', '000000023781', '000000046872', '000000050145','dandelion','rose']
#将自己上传的本地图片名字加入列表中（这里我在原有的基础上添加了'dandelion' and ‘rose’）
show_index = 4# 修改对应索引使用不同图片 0-5

参考文献
[1]：https://developer.huaweicloud.com/develop/aigallery/notebook/detail?id=792c79a5-cc29-4a65-be37-514ab959a692
[2]：https://www.qqtn.com/article/article_276049_1.html

【华为云AI贺新年】有奖征文火热进行中：https://bbs.huaweicloud.com/blogs/325842