正常来说torch模型要转为onnx再转为tensorrt的engine,https://github.com/onnx/onnx-tensorrt这个方法很好,但是注意这个要求的tensorrt的版本是7.1/7.0/6,
然而!妙算2固定jetpack版本3.3,升级之后大疆的osdk不能用,而jetpack和tensorrt的版本是绑定的,jetpack3.3对应的tensorrt版本是4,
所以这件事儿在妙算2上做不成!!!
可以试试在别的机子上转成tensorrt的engine,然后再传到妙算2上调用,不过不知道生成、调用的tensorrt版本不一样能不能用。