grimoire / amirstan_plugin
Useful tensorrt plugin. For pytorch and mmdetection model conversion.
☆165Updated 6 months ago
Alternatives and similar repositories for amirstan_plugin:
Users that are interested in amirstan_plugin are comparing it to the libraries listed below
- A pytorch to tensorrt convert with dynamic shape support☆260Updated last year
- A simple tool that can generate TensorRT plugin code quickly.☆230Updated last year
- TensorRT plugin forDCNv2 layer in ONNX model☆60Updated 4 years ago
- A sample for onnxparser working with trt user defined plugins for TRT7.0☆167Updated 4 years ago
- This project aims to explore the deployment of Swin-Transformer based on TensorRT, including the test results of FP16 and INT8.☆166Updated 2 years ago
- This is a C++ implementation of CenterNet using TensorRT and CUDA☆146Updated 2 years ago
- This repository has been moved. The new location is in https://github.com/TexasInstruments/edgeai-tensorlab☆71Updated last year
- 基于TensorRT7实现DCNv2插件☆48Updated 3 years ago
- Simple demo of tensorrt plugin☆44Updated 3 years ago
- Implement yolov5 with Tensorrt C++ api, and integrate batchedNMSPlugin. A Python wrapper is also provided.☆49Updated 3 years ago
- ⚡ Useful scripts when using TensorRT☆242Updated 4 years ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队:美迪康 AI Lab☆169Updated 2 years ago
- ☆80Updated 4 years ago
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆194Updated 3 years ago
- A tutorial about how to build a TensorRT Engine from a PyTorch Model with the help of ONNX☆243Updated 4 years ago
- A nnie quantization aware training tool on pytorch.☆239Updated 4 years ago
- TensorRT Net Wrapper☆95Updated 5 years ago
- Decode JPEG image on GPU using PyTorch☆90Updated last year
- centernet, mobilenetv2, centerface☆52Updated 5 years ago
- centerNet Caffe inference CPP☆69Updated 5 years ago
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆596Updated last year
- This is 8-bit quantization sample for yolov5. Both PTQ, QAT and Partial Quantization have been implemented, and present the results based…☆102Updated 2 years ago
- CUDA implementation of NMS for PyTorch☆86Updated 5 years ago
- Tensorrt-Deformable-Detr☆59Updated 2 years ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆57Updated 5 years ago
- ☆26Updated 3 years ago
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆160Updated 4 years ago
- Offline Quantization Tools for Deploy.☆127Updated last year
- tensorrt wheels (API)☆77Updated 4 years ago
- Run YoloV3 with the newest TensorRT6.0 at 37 fps on NVIIDIA 1060.☆86Updated 5 years ago