|
|
【人工智能】
rk3399pro 推理精度丢失
发表于 2020-9-24 16:13:03
浏览：6076
|
回复：0
打印
只看该作者
[复制链接]
楼主
在win10 PC机上训练的MTCNN，用pytorch测试无问题，几乎所有目标框中，
模型放在 rk3399pro 上用rknn 推理的时候，几乎无框，检查p网络输出的候选框几乎都没有框中目标。
模型（没有量化和剪枝）是打成onnx 然后转成 rknn 模型，使用 rk3399pro 的 rknn进行的推理。
PC 环境：win10 torch==1.2.0+cpu torchvision= 0.4.0+cpu onnx==1.4.1 numpy==1.16.3 python==3.6.10
rk3399pro ： UBUNTU18.04 numpy==1.16.3 rknn-toolkit==1.3.0 python==3.6.9
请问各位大佬，问题出在哪里？求分享经验 |
|
