HivisionIDPhotos适配昇腾开源验证任务心得

根据任务计划书，主要目的是将HivisionIDPhotos项目适配至华为的 Ascend（昇腾）和 Kunpeng（鲲鹏）处理器，确保其在这硬件平台上的高效运行，也就是说验证HivisionIDPhotos能不能在 Ascend（昇腾）和 Kunpeng（鲲鹏）处理器中运行。

介绍

HivisionIDPhotos 是一款开源的图片处理工具，可以利用 AI 模型对照片进行轻量级智能抠图、调整尺寸生成不同的标准证件照、替换背景、美颜、智能换正装等操作。

开发过程

明确目的后开始熟悉模型，HivisionIDPhotos的github网址：https://github.com/Zeyi-Lin/HivisionIDPhotos。

HivisionIDPhotos项目的使用：参考GitHub项目说明及本开发过程说明

Python版本指定为3.10.0，

conda create -n Photos python=3.10.0
conda activate Photos
cd HivisionIDPhotos

安装项目指定的依赖库：

pip install -r requirements.txt

下载模型权重文件：

python scripts/download_model.py --models all #下载全部模型
# 如需指定下载某个模型
# python scripts/download_model.py --models modnet_photographic_portrait_matting

运行脚本命令示例：

 python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295

在鲲鹏cpu和欧拉操作系统下验证推理

1.安装依赖包

conda create -n Photos python=3.10.0
conda activate Photos
cd HivisionIDPhotos
pip install -r requirements.txt

2.运行脚本命令

 python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295

在NPU运行项目

（1）安装依赖包

使用NPU推理模型需要安装pytorch、CANN和torch_npu，

先安装CANN，参考：https://www.hiascend.com/zh/developer/download/community/result?module=cann。

torch_npu的安装参考：https://github.com/Ascend/pytorch/blob/master/README.zh.md。

其他的库参照开发过程安装。

（2）运行脚本命令：

这个脚本是我修改后的脚本，请访问本项目的gitcode地址（https://gitcode.com/m0_56308334/HivisionlDPhotos/overview）获取：

 python inference_npu.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295

1       测试结果

1.1 CPU运行

运行脚本后输出：

运行后有图片输出：

原图是：                                                       运行脚本后输出一张高清抠图后的4 通道透明一寸证件照格式 png

1.2 Npu运行

1、证件照制作

运行脚本后输出：

原图是：                                                       运行脚本后输出一张高清抠图后的4 通道透明一寸证件照格式 png

2、透明图增加底色

输入 1 张 4 通道透明 png，获得 1 张增加了底色的 3通道图像. 

python inference_2.py -t add_background -i ./idphoto.png -o ./idphoto_ab.jpg  -c 4f83ce -k 30 -r 1  #-r为1 是渐变背景，1为纯色背景。

运行脚本后输出的图片：

显示运行在npu上：