关于Pose Landmark的问题

轰轰轰轰炸鸡 2024-05-22 18:24:21

请教一下各位大佬,snpe支持landmark模型吗?我从mediapipe下载了pose landmark,使用snpe-tflite-to-dlc转模型的时候报错,

File "/home/ts/code/snpe-1.61.0.3358/lib/python/qti/tvm/_ffi/_ctypes/packed_func.py", line 81, in cfun

    rv = local_pyfunc(*pyargs)

File "/home/ts/code/snpe-1.61.0.3358/lib/python/qti/aisw/converters/relay/relay_to_ir.py", line 461, in visit_module

    RelayConverterFrontend.add_op(expr)

File "/home/ts/code/snpe-1.61.0.3358/lib/python/qti/aisw/converters/relay/relay_to_ir.py", line 448, in add_op

    translation.add_op(expr, QUIR_GRAPH, converter_context=CONVERTER_CTX, relay_params=RELAY_PARAMS)

File "/home/ts/code/snpe-1.61.0.3358/lib/python/qti/aisw/converters/relay/translations/relay_translations.py", line 71, in add_op

    input_names)

File "/home/ts/code/snpe-1.61.0.3358/lib/python/qti/aisw/converters/relay/translations/nn_translations.py", line 350, in translate_op

    raise ValueError("Unsupported dynamic weights on tensor {}".format(input_names[1]))

ValueError: Unsupported dynamic weights on tensor permute_0_0

这是不支持吗?链接https://storage.googleapis.com/mediapipe-assets/pose_landmark_heavy.tflite?generation=1661875897944151

...全文
1398 1 打赏 收藏 转发到动态 举报
AI 作业
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
weixin_38498942 2024-05-23
  • 打赏
  • 举报
回复

您好,基于您的报错,问题主要出在snpe不支持dynamic weights。使用netron.app打开您提供的tflite模型,可以发现这是一个量化后的模型,以我框出来的dw卷积为例,这个算子的weight并非常量,而是以location 805的Dequantize算子的输出形式提供的,这种形式就叫dynamic weights。

img

关于Mediapipe的Pose Landmark模型,推荐您尝试这个demo https://aihub.qualcomm.com/iot/models/mediapipe_pose ,这是高通适配过的,可以直接用aihub运行,也可以下载tflite/qnn模型。

3,303

社区成员

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

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