高通AI Hub model GitHub介绍

chipseeker 2024-03-06 16:28:46

前面的文章介绍了高通AI Hub运行环境的配置,高通账号的配置,以及简单的尝试跑了一个ffnet_40例程。

这里我们直接介绍一下高通AI Hub的模型库的开源repo :https://github.com/quic/ai-hub-models

 在model目录下就是AI Hub网页一样能看到所有模型的具体信息,拿unet_segmentation举例,

参考Readme的命令行,就可以尝试把unet跑起来

1. pip3 install "qai-hub-models[unet_segmentation]"

2. python -m qai_hub_models.models.unet_segmentation.demo,在本地PC机跑unet的segmentation的显示,log如下

 

 显示结果,原始图片:

本地分割算法结果:

 

 3.  python -m qai_hub_models.models.unet_segmentation.demo --on-device:在云端的高通手机上跑,log如下

 分割算法结果如下:

 比较本地PC和云端手机的结果略有细微差别,估计是精度不同导致的。后面我们会做精度比较再做分析。

 

 

...全文
1533 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

1,480

社区成员

发帖
与我相关
我的任务
社区描述
本论坛以AI、WoS 、XR、IoT、Auto、生成式AI等核心板块组成,为开发者提供便捷及高效的学习和交流平台。 高通开发者专区主页:https://qualcomm.csdn.net/
人工智能物联网机器学习 技术论坛(原bbs) 北京·东城区
社区管理员
  • csdnsqst0050
  • chipseeker
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧