? ? ? 安裝完TensorRT后砸琅,參考《在Windows10下安裝TensorRT5.1 GA》宋距,要運(yùn)行TensorFlow 模型,必須要先把TensorFlow模型從.pb格式轉(zhuǎn)換為UFF格式症脂。
解決方案:
第一步谚赎,下載TensorRT-5.1.2.2.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.5.tar.gz
TensorRT-5.1.2.2.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.5.tar.gz
第二步:解壓下載文件,找到grashsurgeon和uff文件夾诱篷,依次能找到graphsurgeon-0.4.1-py2.py3-none-any.whl 和?uff-0.6.3-py2.py3-none-any.whl
第三步:用pip install 命令依次安裝graphsurgeon-0.4.1-py2.py3-none-any.whl 和?uff-0.6.3-py2.py3-none-any.whl壶唤。這樣,在Python環(huán)境中棕所,就可以直接用convert-to-uff闸盔,例如:
convert-to-uff frozen_inference_graph.pb -O NMS -p config.py