神经网络pt模型并转换为rknn模型
时间: 2023-04-12 13:00:38 浏览: 619
yolov5_for_rknn:PyTorch中的YOLOv5> ONNX> RKNN
5星 · 资源好评率100%
我可以回答这个问题。将神经网络pt模型转换为rknn模型需要使用Rockchip提供的rknn-toolkit工具。首先,需要将pt模型转换为onnx格式,然后再使用rknn-toolkit将onnx模型转换为rknn模型。具体的转换步骤可以参考rknn-toolkit的官方文档。
阅读全文