wang-xinyu / pytorchx
Implement popular deep learning networks in pytorch, used by tensorrtx.
☆196Updated 3 years ago
Alternatives and similar repositories for pytorchx
Users that are interested in pytorchx are comparing it to the libraries listed below
Sorting:
- Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxn…☆212Updated 3 years ago
- tensorrt int8 量化yolov5 onnx模型☆183Updated 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
- TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据☆526Updated 3 years ago
- tensorrt5, yolov4, yolov3,yolov3-tniy,yolov3-tniy-prn☆264Updated 5 years ago
- tensorrt wheels (API)☆77Updated 4 years ago
- TensorRT ONNX Plugin、Inference、Compile☆468Updated 3 years ago
- A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5☆190Updated 4 years ago
- NVIDIA TensorRT 加速推断教程!☆134Updated 3 years ago
- ☆116Updated last year
- TensorRT int8 量化部署 yolov5s 模型,实测3.3ms一帧!☆168Updated 4 years ago
- Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8☆208Updated last year
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆597Updated last year
- Neural Network Blocks - Collect all kinds of fancy model blocks for you to build more powerful neural network model.☆240Updated 2 years ago
- Got 100fps on TX2. Got 500fps on GeForce GTX 1660 Ti. If the project is useful to you, please Star it.☆178Updated 2 years ago
- ☆711Updated last year
- ⚡ Useful scripts when using TensorRT☆242Updated 4 years ago
- Useful tensorrt plugin. For pytorch and mmdetection model conversion.☆165Updated 7 months ago
- A simple tool that can generate TensorRT plugin code quickly.☆230Updated last year
- A pytorch to tensorrt convert with dynamic shape support☆260Updated last year
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆161Updated 4 years ago
- Implement yolov5 with Tensorrt C++ api, and integrate batchedNMSPlugin. A Python wrapper is also provided.☆49Updated 3 years ago
- ☆129Updated last year
- This repository deploys YOLOv4 as an optimized TensorRT engine to Triton Inference Server☆283Updated 2 years ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队:美迪康 AI Lab☆169Updated 2 years ago
- This is a C++ implementation of CenterNet using TensorRT and CUDA☆146Updated 2 years ago
- QuarkDet lightweight object detection in PyTorch .Real-Time Object Detection on Mobile Devices.☆91Updated 4 years ago
- This is a TensorRT based deepsort project☆79Updated 3 years ago
- yolov5 onnx caffe☆117Updated 4 years ago
- Code for some onnxruntime projects☆122Updated 4 years ago