Toybrick

rknn视频分类模型onnx转rknn

LeonCheung7

新手上路

积分
17
楼主
发表于 2023-9-6 13:48:19    查看: 1331|回复: 1 | [复制链接]    打印 | 只看该作者
请问有人用过视频作为模型输入,在板子上转换为rknn模型的吗?
我在用百度的 ppTSM行为识别模型做动作识别,但是在转换模型时一直报错,算子不支持五维的操作。
我的输入数据是:[1, 8, 3, 224, 224] --> [batch, time, channels , width, height]。
回复

使用道具 举报

LeonCheung7

新手上路

积分
17
沙发
 楼主| 发表于 2023-9-7 15:54:38 | 只看该作者
本帖最后由 LeonCheung7 于 2023-9-7 15:57 编辑

我在加载onnx模型时直接被杀死。
rknn = RKNN(verbose=False)
rknn.config(batch_size=1, target_platform='rk3399pro')
ret = rknn.load_onnx('./weights/ppTSN.onnx')
rknn.release()


(rknn) wict-ai@wictai:~/onnx2rknn$ python model_convert.py
(rknn) wict-ai@wictai:~/onnx2rknn$ python model_convert.py
已杀死

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表