├── .gitignore ├── 1_convert_TF_to_TRT.ipynb ├── 2_inference_using_TensorRT-model.ipynb ├── 3_visualize_the_original_and_optimized_models.ipynb ├── 4_convert_Keras_model_to_Tensorflow.ipynb ├── 5_train_your_own_dataset_in_Keras.ipynb ├── 6_using_trained_model_to_perform_inferece_in_Keras.ipynb ├── 7_optimizing_YOLOv3_using_TensorRT.ipynb ├── README.md ├── YOLOv3 ├── FiraMono-Medium.otf ├── coco.names └── utils.py ├── dataset └── demo_video │ └── road2.mp4 ├── model ├── checkpoint ├── frozen_model.pb ├── modelLeNet5.h5 ├── modelLeNet5_small.h5 └── tensorflow │ └── small │ ├── checkpoint │ ├── model1.data-00000-of-00001 │ ├── model1.index │ ├── model1.meta │ ├── model_small.data-00000-of-00001 │ ├── model_small.index │ └── model_small.meta └── pictures ├── Keras_to_TensorRT.png ├── dog.jpg └── tf-trt_workflow.png /.gitignore: -------------------------------------------------------------------------------- 1 | dataset/ 2 | model/ 3 | .ipynb_checkpoints 4 | __pycache__ 5 | 6 | -------------------------------------------------------------------------------- /1_convert_TF_to_TRT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/1_convert_TF_to_TRT.ipynb -------------------------------------------------------------------------------- /2_inference_using_TensorRT-model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/2_inference_using_TensorRT-model.ipynb -------------------------------------------------------------------------------- /3_visualize_the_original_and_optimized_models.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/3_visualize_the_original_and_optimized_models.ipynb -------------------------------------------------------------------------------- /4_convert_Keras_model_to_Tensorflow.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/4_convert_Keras_model_to_Tensorflow.ipynb -------------------------------------------------------------------------------- /5_train_your_own_dataset_in_Keras.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/5_train_your_own_dataset_in_Keras.ipynb -------------------------------------------------------------------------------- /6_using_trained_model_to_perform_inferece_in_Keras.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/6_using_trained_model_to_perform_inferece_in_Keras.ipynb -------------------------------------------------------------------------------- /7_optimizing_YOLOv3_using_TensorRT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/7_optimizing_YOLOv3_using_TensorRT.ipynb -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/README.md -------------------------------------------------------------------------------- /YOLOv3/FiraMono-Medium.otf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/YOLOv3/FiraMono-Medium.otf -------------------------------------------------------------------------------- /YOLOv3/coco.names: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/YOLOv3/coco.names -------------------------------------------------------------------------------- /YOLOv3/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/YOLOv3/utils.py -------------------------------------------------------------------------------- /dataset/demo_video/road2.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/dataset/demo_video/road2.mp4 -------------------------------------------------------------------------------- /model/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/checkpoint -------------------------------------------------------------------------------- /model/frozen_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/frozen_model.pb -------------------------------------------------------------------------------- /model/modelLeNet5.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/modelLeNet5.h5 -------------------------------------------------------------------------------- /model/modelLeNet5_small.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/modelLeNet5_small.h5 -------------------------------------------------------------------------------- /model/tensorflow/small/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/tensorflow/small/checkpoint -------------------------------------------------------------------------------- /model/tensorflow/small/model1.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/tensorflow/small/model1.data-00000-of-00001 -------------------------------------------------------------------------------- /model/tensorflow/small/model1.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/tensorflow/small/model1.index -------------------------------------------------------------------------------- /model/tensorflow/small/model1.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/tensorflow/small/model1.meta -------------------------------------------------------------------------------- /model/tensorflow/small/model_small.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/tensorflow/small/model_small.data-00000-of-00001 -------------------------------------------------------------------------------- /model/tensorflow/small/model_small.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/tensorflow/small/model_small.index -------------------------------------------------------------------------------- /model/tensorflow/small/model_small.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/model/tensorflow/small/model_small.meta -------------------------------------------------------------------------------- /pictures/Keras_to_TensorRT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/pictures/Keras_to_TensorRT.png -------------------------------------------------------------------------------- /pictures/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/pictures/dog.jpg -------------------------------------------------------------------------------- /pictures/tf-trt_workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardianumam/Tensorflow-TensorRT/HEAD/pictures/tf-trt_workflow.png --------------------------------------------------------------------------------