Unable to convert the saved model to tensorrt.

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

https://github.com/hunglc007/tensorflow-yolov4-tflite

MIT License

2.25k stars 1.24k forks source link

Unable to convert the saved model to tensorrt. #195

Open Whisper94 opened 4 years ago

Whisper94 commented 4 years ago

After running the convert_trt.py is following error

I've tried to find any solution but with tensorflow-gpu 2.3.0 have nothing found. Probably I have wrong installation of TensorRT for Windows? Could you explain how you have integrated TensorRT with tensorflow. Because of tensorflow (also gpu version) couldn't find TensorRT-packages.

Thank you!

Accioy commented 4 years ago

The TensorRT API in Tensorflow is not supported on Windows (you will find it if you take a look the source code).

aniket611 commented 3 years ago

TensorRT python support is not available on windows. Check this link https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#gettingstarted