grimoire / torch2trt_dynamic
A pytorch to tensorrt convert with dynamic shape support
☆258Updated last year
Alternatives and similar repositories for torch2trt_dynamic:
Users that are interested in torch2trt_dynamic are comparing it to the libraries listed below
- Useful tensorrt plugin. For pytorch and mmdetection model conversion.☆163Updated 5 months ago
- ⚡ Useful scripts when using TensorRT☆242Updated 4 years ago
- A tutorial about how to build a TensorRT Engine from a PyTorch Model with the help of ONNX☆243Updated 4 years ago
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆193Updated 3 years ago
- A sample for onnxparser working with trt user defined plugins for TRT7.0☆166Updated 4 years ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队:美迪康 AI Lab☆169Updated 2 years ago
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆594Updated last year
- Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8☆207Updated last year
- volksdep is an open-source toolbox for deploying and accelerating PyTorch, ONNX and TensorFlow models with TensorRT.☆286Updated 4 years ago
- A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5☆189Updated 4 years ago
- A simple tool that can generate TensorRT plugin code quickly.☆228Updated last year
- This project aims to explore the deployment of Swin-Transformer based on TensorRT, including the test results of FP16 and INT8.☆164Updated 2 years ago
- TensorRT ONNX Plugin、Inference、Compile☆466Updated 3 years ago
- ☆79Updated 4 years ago
- This is a C++ implementation of CenterNet using TensorRT and CUDA☆148Updated 2 years ago
- Deploy your model with TensorRT quickly.☆765Updated last year
- TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据☆527Updated 3 years ago
- This repository deploys YOLOv4 as an optimized TensorRT engine to Triton Inference Server☆281Updated 2 years ago
- A nnie quantization aware training tool on pytorch.☆239Updated 4 years ago
- ONNX2Pytorch☆160Updated 3 years ago
- Implement yolov5 with Tensorrt C++ api, and integrate batchedNMSPlugin. A Python wrapper is also provided.☆50Updated 3 years ago
- Decode JPEG image on GPU using PyTorch☆89Updated last year
- 服务侧深度学习部署案例☆451Updated 4 years ago
- Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxn…☆214Updated 3 years ago
- tensorrt int8 量化yolov5 onnx模型☆181Updated 3 years ago
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆160Updated 4 years ago
- ☆710Updated last year
- Simple demo of tensorrt plugin☆43Updated 3 years ago
- caffe model convert to onnx model☆174Updated 2 years ago
- tensorrt wheels (API)☆77Updated 4 years ago