grimoire / mmdetection-to-tensorrt
convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.
☆596Updated last year
Alternatives and similar repositories for mmdetection-to-tensorrt:
Users that are interested in mmdetection-to-tensorrt are comparing it to the libraries listed below
- tensorrt5 , centernet , centerface, deform conv, int8☆768Updated 2 years ago
- TensorRT ONNX Plugin、Inference、Compile☆467Updated 3 years ago
- TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据☆526Updated 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 pytorch to tensorrt convert with dynamic shape support☆260Updated last year
- ☆811Updated 4 years ago
- ☆711Updated last year
- Two-stage CenterNet☆1,216Updated 2 years ago
- ☆487Updated 2 years ago
- Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object Detection, NeurIPS2020☆597Updated 4 years ago
- This repo is implemented based on detectron2 and centernet☆558Updated 4 years ago
- An easy to understand and better performance version of CenterNet☆548Updated 4 years ago
- Deploy your model with TensorRT quickly.☆769Updated last year
- This is the implementation of YOLObile: Real-Time Object Detection on Mobile Devices via Compression-Compilation Co-Design☆368Updated 9 months ago
- detector for rotated-object based on CenterNet/基于CenterNet的旋转目标检测☆400Updated 4 years ago
- reproduce the YOLO series of papers in pytorch, including YOLOv4, PP-YOLO, YOLOv5，YOLOv3, etc.☆392Updated 4 years ago
- tensorrt5, yolov4, yolov3,yolov3-tniy,yolov3-tniy-prn☆265Updated 4 years ago
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆195Updated 3 years ago
- yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.☆729Updated 3 weeks ago
- Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8☆208Updated last year
- Useful tensorrt plugin. For pytorch and mmdetection model conversion.☆165Updated 6 months ago
- A simple pytorch implementation of CenterNet (Objects as Points)☆308Updated 4 years ago
- [CVPR 2020] Detection in Crowded Scenes: One Proposal, Multiple Predictions☆427Updated 3 years ago
- ⚡ Useful scripts when using TensorRT☆242Updated 4 years ago
- You Only Look One-level Feature (YOLOF), CVPR2021, Detectron2☆271Updated 4 years ago
- [CVPR 2020] VoVNet backbone networks for detectron2☆382Updated 3 years ago
- Generalized Focal Loss V2: Learning Reliable Localization Quality Estimation for Dense Object Detection, CVPR2021☆481Updated 3 years ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队：美迪康 AI Lab☆169Updated 2 years ago
- yolov3 with mobilenet v2 and ASFF☆1,066Updated 3 years ago
- Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some…☆783Updated 3 years ago