- 博客(4)
- 收藏
- 关注
原创 pt转onnx转rknn(yolov5部署RK3566)
在终端激活环境,在终端输入pip install -r requirements_cp38-1.5.2.txt -i https://pypi.tuna.tsinghua.edu.cn/simple。然后再输入pip install rknn_toolkit2-1.5.2+708089d1-cp38-cp38-linux_x86_64.whl。该项目需要的yolov5-5.0版本的代码进行训练和模型,最后得到的模型输出头如下图所示。)rknn-toolkit2-1.5.2(笔者的版本)工具进行转换。
2024-12-19 20:21:49
731
原创 RK平台:android上运行NPU和调试NPU
首先,您需要确保您的RK3588开发板已正确连接到您的开发计算机,并且已安装了正确版本的Android开发工具包(Android SDK)和ADB驱动程序。在使用NPU之前,我们需要在应用程序的启动过程中进行NPU的初始化。根据您的模型的要求,您可以创建一个包含输入数据的数据结构,并将其传递给NPU调试工具。请确保你已经准备好了输入数据(例如一个float数组inputData),并根据你的模型的要求进行适当的数据预处理。如果仅进行npu的利用的,建议进行rknn模型的利用。确保您的开发板出现在列表中。
2024-12-19 19:29:53
1681
原创 pt转onnx转ncnn模型(yolov8部署安卓)
我使用的yolov8代码的版本是ultralytics-8.0.50,使用较新的版本会导致模型转换后有很多问题,亲自踩坑。找到Detect下的forward函数这样修改(yolov8——detect这样修改,分割项目不是)编译完成进入build文件夹在这里可以实现转换模型 (找到onnx2ncnn可执行文件)转之前要进行对/ultralytics/nn/modules.py中的代码进行修改。笔者使用的脚本代码进行转换(比较推荐,否则转换ncnn可能有问题)在部署中遇到了若干问题,在此进行记录。
2024-12-19 19:00:54
956
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人