yolo_all_you_need/yolov4/yolov4_python_modelzoo

• 物体检测yolov4 （python）
• • 软件准备
  • 环境部署
  • 运行

物体检测yolov4 （python）

本Application支持运行在Atlas 200 DK上，实现了对黑白图像自动上色的功能。

软件准备

运行此Sample前，需要下载本仓中的源码包。

1. 获取源码包。
   mkdir -p $HOME/AscendProjects

   cd $HOME/AscendProjects

   wget https://c7xcode.obs.cn-north-4.myhuaweicloud.com/code_Ascend/yolov4_python_modelzoo.zip --no-check-certificate

   unzip yolov4_python_modelzoo.zip

   说明：

   • 如果使用wget下载失败，可使用如下命令下载代码。
     curl -OL https://c7xcode.obs.cn-north-4.myhuaweicloud.com/code_Ascend/yolov4_python_modelzoo.zip
   • 如果curl也下载失败，可复制下载链接到浏览器，手动上传至服务器。

2. 从modelzoo下载om模型。

   obs链接：obs://nkxiaolei88/ATC Model/YoloV4/yolov4_no_postprocess.om

3. 将转换好的模型放到工程文件中的model 目录下。
   cp ~/modelzoo/colorization/device/yolov4_no_postprocess.om ~/AscendProjects/yolov4_python_modelzoo/model/

环境部署

1. 应用代码拷贝到开发板。

   以Mind Studio安装用户进入分类网络应用(python)代码所在根目录，如：AscendProjects/classification_python，执行以下命令将应用代码拷贝到开发板。若拷贝失败，请检查开发板上是否有HIAI_PROJECTS这个目录，没有就创建一下。

   scp -r ~/AscendProjects/yolov4_python_modelzoo HwHiAiUser@192.168.1.2:/home/HwHiAiUser/HIAI_PROJECTS

   提示password时输入开发板密码，开发板默认密码为Mind@123

2. acl.so拷贝到开发板。

   scp ~/Ascend/ascend-toolkit/20.0.RC1/arm64-linux_gcc7.3.0/pyACL/python/site-packages/acl/acl.so HwHiAiUser@192.168.1.2:/home/HwHiAiUser/Ascend/

   说明：
   请将X.X.X替换为Ascend-Toolkit开发套件包的实际版本号。
   例如：Toolkit包的包名为Ascend-Toolkit-20.0.RC1-x86_64-linux_gcc7.3.0.run，则此Toolkit的版本号为20.0.RC1。

3. 登录开发板，添加环境变量。

   ssh HwHiAiUser@192.168.1.2
   vim ~/.bashrc
   在最后添加两行
   export LD_LIBRARY_PATH=/home/HwHiAiUser/Ascend/acllib/lib64
   export PYTHONPATH=/home/HwHiAiUser/Ascend/:\${PYTHONPATH}
   
   执行如下命令，使环境变量生效
   source ~/.bashrc

4. 安装环境依赖。

   • 安装numpy、pillow
     请参考 https://gitee.com/ascend/samples/tree/master/common/install_python3env/for_atlas200dk 进行安装。
   • 安装opencv
     请参考 https://gitee.com/ascend/samples/tree/master/common/install_opencv/for_atlas200dk 进行安装。

运行

1. 登录到开发板上，进入工程目录下，执行如下命令运行程序。

   cd ~/HIAI_PROJECTS/yolov4_python_modelzoo/
   python3 colorize.py ./data/

2. 查看推理图片。

   推理产生的结果图片保存在out文件夹，可传到Mindstudio安装用户的家目录中查看。 （此命令不是在开发板一侧使用，而是在MindStudio的安装用户这边使用）

   scp -r username@host_ip:/home/username/HIAI_PROJECTS/yolov4_python_modelzoo/out ~

   • username：开发板用户名，默认为HwHiAiUser。
   • host_ip：开发板ip，USB连接一般为192.168.1.2.网线连接时一般为192.168.0.2。

   命令示例：
   scp -r HwHiAiUser@192.168.1.2:/home/HwHiAiUser/HIAI_PROJECTS/yolov4_python_modelzoo/out ~