rknn视频分类模型onnx转rknn

[复制链接] · 发表于 2023-9-6 13:48:19

请问有人用过视频作为模型输入，在板子上转换为rknn模型的吗？
我在用百度的 ppTSM行为识别模型做动作识别，但是在转换模型时一直报错，算子不支持五维的操作。
我的输入数据是：[1, 8, 3, 224, 224] --> [batch, time, channels , width, height]。

只看该作者 · 发表于 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
已杀死