Toybrick

标题: rknn视频分类模型onnx转rknn [打印本页]

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

作者: LeonCheung7 时间: 2023-9-7 15:54
本帖最后由 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
已杀死

欢迎光临 Toybrick (https://t.rock-chips.com/)