반응형

Deep Learning/TensorRT 4

[TensorRT] Keras, Tensorflow Model을 TensorRT로 변환(TF-TRT Inference)

※ 본 글은 아래의 예제을 test하며, 작성한 글입니다. https://github.com/tensorflow/tensorrt/blob/master/tftrt/examples/image_classification/NGC-TFv2-TF-TRT-inference-from-Keras-saved-model.ipynb GitHub - tensorflow/tensorrt: TensorFlow/TensorRT integration TensorFlow/TensorRT integration. Contribute to tensorflow/tensorrt development by creating an account on GitHub. github.com 1. Env. 아래의 Docker를 사용하였습니다. (colab과 ..

반응형