还在担心黑白老照片没有颜色丨基于ModelArts实例感知图像上色丨【华为云AI贺新年】
前言
ModelArts 是面向开发者的一站式 AI 开发平台，为机器学习与深度学习提供海量数据预处理及交互式智能标注、大规模分布式训练、自动化模型生成，及端-边-云模型按需部署能力，帮助用户快速创建和部署模型，管理全周期 AI 工作流。
前期准备
在使用ModelArts之前，需要进入华为云官网 https://www.huaweicloud.com/ ，然后注册华为云账号，再进行实名认证。主要分为4步（注册–>实名认证–>服务授权–>领代金券）（如有已完成部分，请忽略）
点此去完成 实名认证，账号类型选"个人"，个人认证类型推荐使用"扫码认证"。
具体步骤
下面是给黑白图片上色的步骤。
第一步，点击下方连接进入 实例感知图像上色 的 JupyterLab 页面。
之后配置当前运行环境，进行 规格切换，并选择 [限时免费]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
- 点赞
- 收藏
- 关注作者
评论（0）