GuanLianzheng / pytorch_to_TensorRT5Links
通过pytorch搭建卷积神经网络完成手写识别任务,并将训练好的模型以多种方式部署到TensorRT中加速。(1)通过加载权重。(2)通过加载onnx。(3)通过加载engine。
☆44Updated 6 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
Sorting:
- ☆62Updated 5 years ago
- RepVGG TensorRT int8 量化,实测推理不到1ms一帧!☆62Updated 4 years ago
- A simple demo to use TensorRT with pytorch☆28Updated 6 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
- Implemention of Baidu's DenseBox used for multi-task learning of object detection and landmark(key-point) localization 用PyTorch实现了百度的Den…☆95Updated 4 years ago
- this repo is forked from https://github.com/amdegroot/ssd.pytorch. Implemented by pytorch☆46Updated 5 years ago
- Code for some onnxruntime projects☆122Updated 4 years ago
- Convert YOLOv3 and YOLOv3-tiny (PyTorch version) into TensorRT models.☆61Updated 5 years ago
- 一个简单的TensorRT实现语义分割的C++程序☆60Updated 5 years ago
- EfficientNetv2 TensorRT int8☆46Updated 4 years ago
- centerNet Caffe inference CPP☆68Updated 6 years ago
- CCFDF AI 数钢筋大赛☆172Updated 6 years ago
- 重新设计的RFBNet300,模型参数量只有0.99MB,AP达到0.78,200FPS☆100Updated 5 years ago
- ☆27Updated 6 years ago
- 数钢筋demo,IOU 0.7 下,AP 90.6。训练只要不到十分钟,可以非常愉快的 玩耍☆33Updated 5 years ago
- A C++ inference SDK based on TensorRT☆70Updated 4 years ago
- You Only Look At Face☆35Updated 5 years ago
- CenterNet's inference.(C++)/基于CenterNet的旋转目标检测C++版☆28Updated 4 years ago
- LFFD and SSD based Method implemented by Pytorch for Basketball Recognition Competition.☆34Updated 5 years ago
- TensorRT inference sample for ResNet50☆34Updated 6 years ago
- A real-time version of FCOS, inspired by FCOSv2.☆49Updated 3 years ago
- mmdetection in python训练, libtorch in c++部署☆12Updated 4 years ago
- MobileNet v3☆49Updated 6 years ago
- Examples and tools for deep learning deployment☆57Updated 4 years ago
- You can import this module directly☆55Updated 6 years ago
- centernet_mobilenetv2 inference by ncnn☆65Updated 5 years ago
- Object detection model thundernet tensorflow.keras implementation☆83Updated 6 years ago
- pytorch 使用centernet进行 人脸检测Face Detection (bbox + landmark)、 船只检测SeaShips Detection☆48Updated 5 years ago
- To augment image dataset for detection☆58Updated 6 years ago
- 基于CenterNet训练的目标检测&人脸对齐&姿态估计模型☆283Updated 4 years ago