将PB模型转换为tensorRT引擎的uff文件的问题

时间:2020-07-24 01:42:53

标签: nvidia automl tensorrt uff

  1. 我们正在尝试将使用Google AutoML Vision(https://cloud.google.com/vision/automl/object-detection/docs/export-edge#deploy-example-web)训练的AutoML Edge .pb模型转换为.uff文件,以便我们可以使用TensorRT引擎执行推理。
  2. 我们在将冻结的.pb转换为.uff时遇到了上述问题(https://imgur.com/bhyXjh4
  3. 设置详细信息:Ubuntu 18.04,TensorRT-7.0.0.11,Quadro P1000 GPU

有人解决过这个问题吗?

0 个答案:

没有答案