pytorch模型转为rknn后没有推理结果

715 人工智能 pytorch RK3566

问答对人有帮助，内容完整，我也想知道答案 0 模型输入为[1, 24, 100, 5] 输出为[3] 目标平台为rk3566 我通过训练时得到的.pth保存模型参数并加载模型转为.pt文件模型中大部分都是2d的Conv层+leakyRelu的组合使用rknn的api读取我的模型，进行转换 api可以成功转换出rknn模型，但遇到的问题是：我使用测试数据调用 `rknn.inference` 进行推理，每一次的输出结果都为[array([nan, nan, nan], dtype=float32)] 如果我在 `rknn.build` 的阶段，对模型进行量化，那每次都可以输出结果，但不同的输入，都会获得一样的结果。我百思不得其解，在网上也没有人遇到相同的问题，希望有大神能为我解答，非常非常感谢！！ 0 已退回1积分
2023-1-11 18:45:48　　评论淘帖0 邀请回答您可以邀请以下用户，快速回答问题 × liutiefu 该类别下有 26 个回答。邀请回答 ze55me 该类别下有 20 个回答。邀请回答 tigerwang711 该类别下有 20 个回答。邀请回答杀狼000 该类别下有 19 个回答。邀请回答 zhuzb0754 该类别下有 19 个回答。邀请回答冰箱洗衣机该类别下有 18 个回答。邀请回答 YYXIAO 该类别下有 18 个回答。邀请回答河神大人该类别下有 17 个回答。邀请回答 mede1001 该类别下有 17 个回答。邀请回答安德森大该类别下有 17 个回答。邀请回答 CDCNKA 该类别下有 17 个回答。邀请回答 jjll652 该类别下有 17 个回答。邀请回答熊本熊该类别下有 16 个回答。邀请回答 YOYOOO 该类别下有 16 个回答。邀请回答麻酱该类别下有 16 个回答。邀请回答世态薄凉该类别下有 16 个回答。邀请回答 musicalboy 该类别下有 15 个回答。邀请回答万航渡路该类别下有 15 个回答。邀请回答 wuli北该类别下有 15 个回答。邀请回答五斤麻辣油该类别下有 15 个回答。邀请回答举报 jf_94992931 相关推荐 • rv1126 进行yolov5视觉项目的推理精度下降 1805 • 如何去使用RK3566内置NPU模块呢 10931 • RKNN-Toolkit运行非 RKNN 模型时工具的使用流程是怎样的 2499 • 怎样去解决pytorch模型一直无法加载的问题呢 2847 • 为什么RK3399Pro平台YOLOv4 pytorch模型转RKNN模型会失败呢 1898 • YOLOv3的darknet模型先转为caffe模型后再转为fp32bmodel，模型输出和原始模型输出存在偏差是怎么回事？ 225 • 如何提高YOLOv4模型的推理性能？ 895 • 请问一下rknn多图推理参数该怎样去设置呢 2978 • RK3566（rknn-toolkit2) 转换facenet失败是怎么回事 2201 • 请问在rknn_yolov5_demo中batch设置和推理时间长短有何关系 5676