还在担心黑白老照片没有颜色丨基于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)