grimoire / amirstan_plugin
Useful tensorrt plugin. For pytorch and mmdetection model conversion.
☆159Updated last month
Related projects ⓘ
Alternatives and complementary repositories for amirstan_plugin
- A pytorch to tensorrt convert with dynamic shape support☆257Updated 9 months ago
- A sample for onnxparser working with trt user defined plugins for TRT7.0☆166Updated 4 years ago
- A simple tool that can generate TensorRT plugin code quickly.☆221Updated last year
- This is a C++ implementation of CenterNet using TensorRT and CUDA☆149Updated last year
- Simple demo of tensorrt plugin☆42Updated 3 years ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团 队:美迪康 AI Lab☆164Updated 2 years ago
- Implement yolov5 with Tensorrt C++ api, and integrate batchedNMSPlugin. A Python wrapper is also provided.☆50Updated 3 years ago
- ☆78Updated 4 years ago
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆191Updated 2 years ago
- This project aims to explore the deployment of Swin-Transformer based on TensorRT, including the test results of FP16 and INT8.☆161Updated last year
- A tutorial about how to build a TensorRT Engine from a PyTorch Model with the help of ONNX☆243Updated 3 years ago
- 基于TensorRT7实现DCNv2插件☆48Updated 2 years ago
- ⚡ Useful scripts when using TensorRT☆240Updated 4 years ago
- A nnie quantization aware training tool on pytorch.☆238Updated 3 years ago
- TensorRT plugin forDCNv2 layer in ONNX model☆58Updated 4 years ago
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆589Updated 9 months ago
- TensorRT ONNX Plugin、Inference、Compile☆465Updated 3 years ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆57Updated 5 years ago
- TensorRT Net Wrapper☆95Updated 5 years ago
- CUDA implementation of NMS for PyTorch☆85Updated 4 years ago
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆158Updated 3 years ago
- Offline Quantization Tools for Deploy.☆116Updated 10 months ago
- tensorrt int8 量化yolov5 onnx模型☆176Updated 3 years ago
- ☆39Updated 2 years ago
- ☆23Updated 3 years ago
- This repository has been moved. The new location is in https://github.com/TexasInstruments/edgeai-tensorlab☆70Updated 7 months ago
- TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据☆523Updated 3 years ago
- A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5☆190Updated 4 years ago
- This repository deploys YOLOv4 as an optimized TensorRT engine to Triton Inference Server☆279Updated 2 years ago