|
我也是用Ultra-Light-Fast-Generic-Face-Detector-1MB,这个的RFB-640的模型,我做了相应的裁剪,然后也用onnxsim做了模型简化,但是当我把简化的模型转为rknn的时候就报错了:
IndexError:_Map_base::at
我百度了,说是我的torch.onnx.export需要个 keep_initializers_as_inputs 这个参数,我现在的版本根本就没有这个参数,我的就是pytorch1.2.0直接转的onnx1.4.1的,所以不知道该怎么继续,如果你也有这个经历,麻烦告知,谢谢 |
|