mingxingtan
commented
3 years ago Could you try this script: https://github.com/google/automl/blob/master/efficientdet/tensorrt.py?
Open tricoffee opened 3 years ago
mingxingtan
commented
3 years ago Could you try this script: https://github.com/google/automl/blob/master/efficientdet/tensorrt.py?
I follow this step . Export SavedModel, frozen graph, tensort models, or tflite.
Run the following command line to export models:
!rm -rf savedmodeldir !python model_inspect.py --runmode=saved_model --model_name=efficientdet-d0 \ --ckpt_path=efficientdet-d0 --saved_model_dir=savedmodeldir \ --tensorrt=FP32 --tflite_path=efficientdet-d0.tflite
but it occurs an error:
W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer.so.6'; dlerror: /lib/x86_64-linux-gnu/libm.so.6: version `GLIBC_2.27' not found (required by /usr/lib/libnvinfer.so.6); LD_LIBRARY_PATH: /usr/local/nvidia/lib:/usr/local/nvidia/lib64