mmdetection-to-tensorrt icon indicating copy to clipboard operation
mmdetection-to-tensorrt copied to clipboard
verified publisher icon grimoire

what is the difference between mmdetection-to-tensorrt and torch2trt

Open liuhuiCNN opened this issue 4 years ago • 1 comments

Describe the bug RT, what is the difference between mmdetection-to-tensorrt and torch2trt (https://github.com/NVIDIA-AI-IOT/torch2trt)?

liuhuiCNN avatar May 24 '21 09:05 liuhuiCNN

Errr... mmdetection-to-tensorrt focus on convert object detection and instance segmentation model in mmdetection(only) to TensorRT, torch2trt is a more general tool for PyTorch to TensorRT. Actually, mmdetection-to-tensorrt is based on torch2trt_dynamic, which is once a folk of torch2trt with dynamic shape support.

grimoire avatar May 24 '21 14:05 grimoire