rk3399pro yolov5 NPU加速
时间: 2023-12-06 19:37:39 浏览: 114
yolov5的TensorRT模型加速资源
根据提供的引用内容,介绍rk3399pro yolov5 NPU加速的过程如下:
1. 在服务器端或电脑端进行训练,训练完成后将.pt文件转换成ONNX格式。
2. 使用rknn-toolkit1.6.0将ONNX模型转换成RKNN模型。
3. 在RK3399Pro中进行模型推理,使用NPU加速。
具体的流程和踩坑记录可以参考提供的两个引用内容。需要注意的是,NPU加速需要硬件支持,因此需要使用支持NPU的硬件平台,如RK3399Pro。另外,rknn-toolkit1.6.0是一个用于将ONNX模型转换成RKNN模型的工具,需要在电脑端安装使用。
阅读全文