search

grimoire / mmdetection-to-tensorrt

convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.
Apache License 2.0
585 stars 85 forks source link

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

Open liuhuiCNN opened 3 years ago

liuhuiCNN commented 3 years ago

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

grimoire commented 3 years ago

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.