在小藤上实现Yolact实例分割
【摘要】 在小藤上实现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)