rmccorm4 / tensorrt-utils

⚡ Useful scripts when using TensorRT
Apache License 2.0
240 stars 55 forks source link

int8 engine for **DETECTOR models(onnx) #25

Open Abrahamma97 opened 4 years ago

Abrahamma97 commented 4 years ago

Description

I have seen examples where classification models can be run on TensorRT in INT8 mode. But can you get specific on what I should do to calibrate the same and produce int8 engine for DETECTOR models(onnx)

Environment

TensorRT Version:7 GPU Type: T4 Nvidia Driver Version: 440 CUDA Version: 10.2 CUDNN Version: Operating System + Version: 18 Python Version (if applicable): TensorFlow Version (if applicable): PyTorch Version (if applicable): Baremetal or Container (if container which image + tag):

Relevant Files

Steps To Reproduce