zhangming8 / pytorch-cpp-modelLinks
use c++ load pytorch model and use GPU to predict
☆28Updated 6 years ago
Alternatives and similar repositories for pytorch-cpp-model
Users that are interested in pytorch-cpp-model are comparing it to the libraries listed below
Sorting:
- 通过pytorch搭建卷积神经网络完成手写识别任务，并将训练好的模型以多种方式部署到TensorRT中加速。（1）通过加载权重。（2）通过加载onnx。（3）通过加载engine。☆44Updated 6 years ago
- 对https://github.com/ruinmessi/RFBNet 代码做中文注释,便于理解☆21Updated 6 years ago
- Pytorch libtorch demo☆35Updated last year
- 总结-改进-汇总☆145Updated 5 years ago
- 一个简单的TensorRT实现语义分割的C++程序☆60Updated 6 years ago
- Code for some onnxruntime projects☆121Updated 4 years ago
- A simple demo to use TensorRT with pytorch☆28Updated 6 years ago
- To augment image dataset for detection☆58Updated 6 years ago
- RetinaNet with backbone se_resnext50_32x4d, se_resnext101_32x4d☆25Updated 6 years ago
- ☆62Updated 5 years ago
- A module to convert pytorch model to caffe model.☆67Updated 5 years ago
- CCFDF AI 数钢筋大赛☆172Updated 6 years ago
- A bite of cpp api in PyTorch-1.0☆163Updated 5 years ago
- tensorflow implementation for Yolo v4☆60Updated 3 years ago
- Minimal PyTorch implementation of YOLOv3☆61Updated 6 years ago
- MobileNet v3☆49Updated 6 years ago
- EfficientNetv2 TensorRT int8☆46Updated 4 years ago
- Implemention of Baidu's DenseBox used for multi-task learning of object detection and landmark(key-point) localization 用PyTorch实现了百度的Den…☆95Updated 5 years ago
- RepVGG TensorRT int8 量化，实测推理不到1ms一帧！☆62Updated 4 years ago
- The use examples of tensorboard on pytorch☆148Updated 7 years ago
- 深度学习/计算机视觉工程向笔记☆11Updated 6 years ago
- A flexible Python front-end inference SDK based on TensorRT☆87Updated 4 years ago
- YOLO系列资料☆50Updated 3 years ago
- 数钢筋demo，IOU 0.7 下，AP 90.6。训练只要不到十分钟，可以非常愉快的 玩耍☆33Updated 6 years ago
- You can import this module directly☆55Updated 6 years ago
- Hybrid_Task_Cascade in mmdetection☆20Updated 6 years ago
- DCIC 钢筋数量AI识别 baseline 0.98+。☆71Updated 4 years ago
- Choose the classes you want from some public datasets and make it into voc format☆28Updated 6 years ago
- A resnet18 version of CenterNet（objects as points）☆124Updated 4 years ago
- ☆23Updated 6 years ago