pt转rknn出错

首先你得先用pytorch1.2 加载这个模型，能正确推理了，再转rknn

不知道你什么意思，你都说用rknn可以推理成功了，那还有什么问题么？

zhw 发表于 2020-12-30 09:41
torch1.2可以正常推理成功，但是转rknn出错，实在是找不到问题所在了

你把你torch1.2 读取这个pt推理正确的代码贴出来看下

zhw 发表于 2020-12-30 09:47
device = torch.device("cpu")
weights = 'plate.pt'
model = torch.load(weights, map_location=device) ...

为何需要['model'].eval()
直接eval的模型无法推理么？

zhw 发表于 2020-12-30 10:02
eval不是转换成推理模式吗？我去掉.eval()也可以推理成功

我是说['model']

zhw 发表于 2020-12-30 10:10
model = torch.load(weights, map_location=device).eval()
AttributeError: 'dict' object has no a ...

说明你的模型并不是通用模型。
rknn也是得先读取torch模型才能转换的，都不能用常规方法读出来，转换肯定会出问题。

pytorch我并不是很熟，还是建议你qq群里问下NPU部门的人，你这个问题解决后如果还不行
可以试着把模型文件@给他们让他们调试下

zhw 发表于 2020-12-30 10:31
weights = 'plate.pt'
model = torch.load(weights, map_location=device)['model'].float().eval()
torc ...

群管理员都是