我遇到这个问题,在windows训练了yolov5的pt模型转换成onnx以后去转rknn模型,转换出来的模型在用图片进行测试的时候识别度下降了快一半,经过一个星期的努力解决了,最后发现是我的推理测试代码有问题,
我的解决思路是:
用https://github.com/airockchip/rknn_model_zoo下面的python例子用你转换好的rknn 或者onnx 模型连接板子进行推理测试,也可以用c的,但是python的简单点不需要修改什么文件,c的话需要修改配置文件,有点麻烦,看结果如果不理想可能就是你的模型有问题
yolo5训练的模型在windos识别度高 但是转rknn以后识别度下降
最新推荐文章于 2024-06-21 22:27:17 发布