GuanLianzheng / pytorch_to_TensorRT5
通过pytorch搭建卷积神经网络完成手写识别任务，并将训练好的模型以多种方式部署到TensorRT中加速。（1）通过加载权重。（2）通过加载onnx。（3）通过加载engine。
☆45Updated 5 years ago
Alternatives and similar repositories for pytorch_to_TensorRT5:
Users that are interested in pytorch_to_TensorRT5 are comparing it to the libraries listed below
- RepVGG TensorRT int8 量化，实测推理不到1ms一帧！☆62Updated 3 years ago
- ☆62Updated 4 years ago
- A simple demo to use TensorRT with pytorch☆28Updated 5 years ago
- Examples and tools for deep learning deployment☆56Updated 4 years ago
- ☆79Updated 4 years ago
- This is the implementation that supports yolov5s, yolov5m, yolov5l, yolov5x.☆34Updated 2 years ago
- centernet_mobilenetv2 inference by ncnn☆64Updated 5 years ago
- YOLOV3-Tiny TensorRT6.0 13个类别☆32Updated 4 years ago
- this repo is forked from https://github.com/amdegroot/ssd.pytorch. Implemented by pytorch☆46Updated 5 years ago
- ☆25Updated 5 years ago
- 一个简单的TensorRT实现语义分割的C++程序☆58Updated 5 years ago
- U版yolov5 2.0的tensorrt加速☆37Updated 4 years ago
- ☆14Updated 5 years ago
- YOLOv3-training-prune☆59Updated 3 years ago
- Convert YOLOv3 and YOLOv3-tiny (PyTorch version) into TensorRT models.☆61Updated 5 years ago
- pytorch 使用centernet进行 人脸检测Face Detection (bbox + landmark)、 船只检测SeaShips Detection☆47Updated 5 years ago
- 对https://github.com/ruinmessi/RFBNet 代码做中文注释,便于理解☆20Updated 6 years ago
- A flexible Python front-end inference SDK based on TensorRT☆85Updated 3 years ago
- ☆26Updated 3 years ago
- This is the unofficial implementation of the "CenterNet:Objects as Points".Just a simple try with self-modified shufflenetv2 and yolov3.…☆111Updated 3 years ago
- Code for some onnxruntime projects☆122Updated 3 years ago
- NVIDIA TensorRT 加速推断教程！☆133Updated 3 years ago
- provide pytorch model and ncnn model☆76Updated 4 years ago
- 使用pytorch实现mtcnn的训练及预测，并用pytorch的c++ api实现了前向预测的功能化☆39Updated 4 years ago
- centerNet Caffe inference CPP☆69Updated 5 years ago
- A C++ inference SDK based on TensorRT☆70Updated 4 years ago
- Implemention of Baidu's DenseBox used for multi-task learning of object detection and landmark(key-point) localization 用PyTorch实现了百度的Den…☆94Updated 4 years ago
- TensorRT inference sample for ResNet50☆34Updated 6 years ago
- Apply the pruning strategy for MobileNet_v2☆51Updated 5 years ago
- 重新设计的RFBNet300,模型参数量只有0.99MB，AP达到0.78，200FPS☆99Updated 5 years ago