【人工智能平台实践】华为Atlas200dk之黑白照片上色

        这是我们的课程设计项目,前面走了很多弯路,说不定未来有同学也要做这个,所以记录一下。首先大致描述一下项目流程,然后主要详细记录了环境搭建和硬件连接的步骤,最后附上了实验报告的部分详细内容(ATC模型转换详细讲解)。

        老师在课程群里会发Atlas200dk环境部署的pdf文档教程,还要求下载VMware和Ubuntn虚拟机,用虚拟机做项目。我们组用了四周的时间研究虚拟机的搭建、换源、Mindstudio等,但是总是不能成功,后来我们发现环境合设可以直接在Windows系统上做,所以我们放弃了虚拟机。

         首先清sd卡,烧录环境镜像(如果上一届做过这一步,可以省略),然后按步骤进行设置。完成后即可进入CANN 的samples库,选择自己想做的代码。代码库里的步骤十分详细,可以直接上手。

https://gitee.com/ascend/samples/tree/master/python/level2_simple_inference/6_other/colorization_pictureicon-default.png?t=N658https://gitee.com/ascend/samples/tree/master/python/level2_simple_inference/6_other/colorization_picture

1.项目选题

随着技术的不断发展,黑白照片上色已经成为了一种热门的图像处理技术。黑白照片可以通过上色得到更加真实的视觉效果,也可以帮助人们更好地了解历史。但是,有效地给黑白照片上色并不是一件容易的事情。在本文中,我们将探讨黑白照片上色方法有哪些,以及它们的优缺点和适用场景。

2.系统软硬件环境搭建

2.1部署准备

1.基本准备

Atlas 200 DK开发者套件(USB-Type-C数据线和网线、16G以上SD卡、读卡器或母对母杜邦线)

Windows系统PC机(本项目运行环境与开发合设可以使用windows系统)

MobaXterm(合设环境的情况下,使用MobaXterm便于访问200dk)

balenaEtcher(合设环境必备,使用balenEtcher工具部署合设环境)

2.安全清空sd卡

ps:用读卡器很容易报错,不能成功清空和成功烧入,最好能直接把sd卡插电脑里

步骤如下:

使用读卡器将sd卡连接到电脑,win+R->运行->输入cmd->在命令提示符窗里输入“DISKPART”

 

在弹出的DISKPART窗口中输入“LIST DISK”,按下回车后会显示电脑上的硬盘和U盘。根据磁盘空间大小可以推断出磁盘1为sd卡。

 

首先输入SELECT DISK 1,选中磁盘1;

接着依次执行以下命令:

CLEAN:删除整个磁盘数据。

CREATE PARTITION PRIMARY:创建主磁盘分区。

ACTIVE:激活磁盘分区。

FORMAT QUICK:快速格式化磁盘分区。

 

如图所示,sd卡已制好,可以开始烧录。

2.2搭建运行环境

本项目选择将Atlas 200 DK即作为开发环境又作为运行环境也就是采取合设的方式搭建环境。我们通过dd镜像的方式来将环境恢复到sd卡中再升级至需要的版本使用合设,PC机甚至不需要安装Ubuntu系统就可以在200DK上进行开发

  1. 下载Etcher工具、sd卡镜像
  2.  这是镜像的链接:https://bbs.huaweicloud.com/forum/thread-139685-1-1.htmlicon-default.png?t=N658https://bbs.huaweicloud.com/forum/thread-139685-1-1.html

 

 

解压镜像后,打开Etcher工具,选择img文件和sd卡设备,点击Flash,一直等到Flash成功,大概需要20分钟左右。

拔出sd卡,将其插入Atlas200DK开发板中,上电启动,开发板会进行自动升级固件、重启灯动作,大概需要等待5分钟左右,四个led灯全亮,至此大功告成

 

2.3.开发板连接PC机

Atlas 200 DK开发者板支持通过USB端口或者网线与PC机进行连接
Atlas 200 DK与PC机连接图:

需要配置好USB端口连接,才能配置网线连接

1.USB端口连接
  • 使用套件内的黑色USB-Type-C数据线连接开发板和PC机

PC机打开设备管理器->其他设备,可以看到带有警报标志的RNDIS, 右键点击RNDIS->更新驱动程序

 

选择“浏览我的电脑以查找驱动程序”

 

选择“让我从计算机上的可用驱动程序列表中选取”

 

向下滑动,选择“网络适配器”,然后下一页

 

厂商选择Microsoft,型号选择USB RNDIS6 适配器,更新驱动程序完成进入网络和Internet->高级网络设置->更多网络适配器选项,可以查看到USB RNDIS6 适配器,右键点击查看属性,双击ipv4协议选择”使用下面的ip地址“,将ip地址更改为192.168.1.X, X为0~255除了2以外的数字,点击子网掩码填写框会自动生成子网掩码,点击确定

 

PC机打开cmd终端,输入ssh HwHiAiUser@192.168.1.2按下Enter键,连接成功。

2.网线连接
  • 使用套件内的网线和USB-Type-C数据线连接开发板和PC机
  • 连接开发板,输入su,然后回车,进入root模式,执行cd命令进入netplan文件夹,并查看文件夹下文件.使用vi命令打开网络配置文件并做出修改,修改后的结果如图所示。

 

 

使用netplan apply命令,使设置的网络参数生效重启开发板连接进入,ping百度网测试,如图则表示联网成功。

 下面就是去昇腾sample库里找合适的代码。注意辨别型号,还有一些题目需要先一键制卡,我们组一键制卡失败了,所以没有记录过程。

4.2 ATC模型转换(这个在代码库里有,直接运行就可以,其实不用自己写,这个是详细解释,写在报告里的)

昇腾张量编译器(Ascend Tensor Compiler,简称ATC)是昇腾CANN架构体系下的模型转换工具它可以将开源框架的网络模型(例如TensorFlow、ONNX等)转换为昇腾AI处理器支持的模型文件(.om格式),用于后续的模型推理。

ATC工具功能架构

 

本项目所用atc

--input_shape="data_l:1,1,224,224"

--weight="./colorization.caffemodel"

--input_format=NCHW

--output="./colorization_yuv"

--soc_version=Ascend310

--framework=0

--model="./colorization.prototxt"

解释:对模型colorization.prototxt和权重colorization.caffemodel的caffe原始模型进行转换,因此framework为0表示caffe,输出文件名字为colorization_yuv,推理芯片选择Ascend310

常用参数解释:

--model:原始模型文件路径与文件名

--weight:权重文件路径与文件名

--framework:原始模型框架:0是caffe,1是mindspore,3是tensorflow,5是onnx

--input_format:输入数据格式:Caffe默认为NCHW,TensorFlow默认为NHWC

--dynamic_image_size:设置输入图片的动态分辨率参数。适用于执行推理时,每次处理图片宽和高不固定的场景。

--output:存放转换后的离线模型的路径以及文件名

--soc_version:模型转换时指定芯片版本。

--insert_op_conf:插入算子的配置文件路径与文件名,例如aipp预处理算子(包含了图像预处理比如色域转换、减均值等)。

若提示ATC run success信息,则说明模型转换成功。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值