rknn-toolkit装了好多次,总是会报错,终于尝试出了一个版本可以用的。(注意windows和linux上都能用)
rknn-toolkit: v1.7.1:
cuda 10.1
python==3.6
pytorch==1.5.1
opencv-python==3.2.0.8
tensorflow==1.14.0
模型导出的时候,最好pytorch版本也是1.5.1,导出的代码是:
trace_model = torch.jit.trace(model.model, torch.rand(1, 3, 32, 38).cuda())
torch.jit.save(trace_model, "./rknn_ocr.pt")
如果不行的话自己先试一试,可视化里面导入的代码是:
model = torch.jit.load('./rknn_ocr.pt')