在小藤上实现Yolact实例分割

举报
张辉 发表于 2023/09/01 07:27:23 2023/09/01
【摘要】 在小藤上实现Yolact实例分割
近期小藤的官方刻录工具推出了带有桌面版的镜像:

完成刻录后,将其插入小藤的SD卡槽,启动系统并登录,可以看到环境如下:
操作系统:Ubuntu 22.04
CANN:6.2.RC2

MindX:mxVision 5.0.RC2

Python:conda环境,base为3.9.2

这个镜像已经做好了swap分区(8G还挺大的)。所以不需要每次启动的时候改分区了。

先安装opencv-python PIP依赖包:
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

设置CANN和MindX的环境变量:
source /usr/local/Ascend/mxVision/setenv.sh
source /usr/local/Ascend/ascend-toolkit/set_env.sh

尴尬了,没有权限。

切换到root,给ascend-toolkit和mxVision目录增加权限:

chmod 777 /usr/local/Ascend/mxVision-5.0.RC2/set_env.sh
chmod 777 /usr/local/Ascend/ascend-toolkit/set_env.sh

返回到HwHiAiUser用户:
source /usr/local/Ascend/mxVision-5.0.RC2/set_env.sh
source /usr/local/Ascend/ascend-toolkit/set_env.sh

下载ascend_community_projects代码仓的310B分支:
git clone https://gitee.com/ascend/ascend_community_projects -b 310B

cd ascend_community_projects/Yolact
cd convert
下载onnx模型:
wget https://mindx.sdk.obs.cn-north-4.myhuaweicloud.com/ascend_community_projects/Yolact/models.onnx --no-check-certificate
执行模型转换:
chmod +x convert_om.sh

./convert_om.sh models.onnx models
模型转换的时间比较长,要耐心等待。

可以看到om文件已经生成。
找一张图片:test000.png

传到小藤的 ascend_community_projects/Yolact/input目录:

vi ./model_data/yolact.pipeline 中16行中修改modelPath,指向om模型文件所在路径:

cd ../
python3 main.py --image='input/test000.png'

经过 “路过一小只”的提醒,应该也是权限问题:


执行以下操作:

再试试:

原来so也要权限:

再来:

再试试:

查看权限:

看来应该把整个mxVision目录权限都改掉的:

再来:


看来还得操作全部目录:

再来:


再继续改改:


感觉陷入了一个怪圈,权限不足的话打不开文件,权限太多的话,程序又报错:
GST_PLUGIN_SCANNER permission is higher than 0550
这个时候有2种解决方案:
1.权限不改,代码使用root执行。
2.修改文件属主为HwHiAiUser,并把文件权限改到代码满意为止。

有没有第三种方法呢?
那就是用普通用户重新装一下MindX SDK吧!
打开


下载 aarch64的mxVision版本:

将其传到小藤上:

先卸载试试:
chmod +x *.run
./Ascend-mindxsdk-mxvision_5.0.RC2_linux-aarch64.run --check

./Ascend-mindxsdk-mxvision_5.0.RC2_linux-aarch64.run --install

当前包安装到了 ~/mxVision目录下了:

MindX的环境变量重新执行下:
source /home/HwHiAiUser/mxVision-5.0.RC2/set_env.sh

cd ascend_community_projects/Yolact
python3 main.py --image='/home/HwHiAiUser/ascend_community_projects/Yolact/input/test000.png'

看来png的格式不行,需要改为jpg格式:(使用画图转一下)

python3 main.py --image='/home/HwHiAiUser/ascend_community_projects/Yolact/input/test000.jpg'

将结果图片下载到本地:

感觉中间两个人识别的有点错位。
换张图片试试:test001.jpg


结果文件仍然是那个文件名,下载打开看看:

感觉模型精度有待提高啊。。

(全文完,谢谢阅读)
【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。