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:
- CCFDF AI 数钢筋大赛☆172Updated 6 years ago
- RetinaNet with backbone se_resnext50_32x4d, se_resnext101_32x4d☆25Updated 7 years ago
- 通过pytorch搭建卷积神经网络完成手写识别任务,并将训练好的模型以多种方式部署到TensorRT中加速。(1)通过加载权重。(2)通过加载onnx。(3)通过加载engine。☆44Updated 6 years ago
- DCIC 钢筋数量AI识别 baseline 0.98+。☆71Updated 4 years ago
- 总结-改进-汇总☆145Updated 5 years ago
- Pytorch libtorch demo☆35Updated 2 years ago
- 一个简单的TensorRT实现语义分割的C++程序☆59Updated 6 years ago
- ☆62Updated 5 years ago
- To augment image dataset for detection☆58Updated 6 years ago
- Code for some onnxruntime projects☆120Updated 4 years ago
- DF-Traffic-Sign-Identification☆30Updated 6 years ago
- 深度学习/计算机视觉工程向笔记☆11Updated 6 years ago
- A flexible Python front-end inference SDK based on TensorRT☆87Updated 4 years ago
- 对https://github.com/ruinmessi/RFBNet 代码做中文注释,便于理解☆21Updated 7 years ago
- A bite of cpp api in PyTorch-1.0☆163Updated 5 years ago
- 深度学习500问,以问答形式对常用的概率知识、线性代数、机器学习、深度学习、计算机视觉等热点问题进行阐述,以帮助自己及有需要的读者。 全书分为18个章节,近30万字。由于水平有限,书中不妥之处恳请广大读者批评指正。 未完待续............ 如有意合作,联系sc…☆54Updated 7 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
- 实现常用的one-stage和two-stage目标检测网络☆54Updated 7 years ago
- MobileNet v3☆49Updated 6 years ago
- A simple demo to use TensorRT with pytorch☆28Updated 6 years ago
- simple rebar detection competition https://www.datafountain.cn/competitions/332/details☆38Updated 7 years ago
- You can import this module directly☆55Updated 6 years ago
- the speed of nms☆48Updated 7 years ago
- cv的一些比赛☆45Updated 6 years ago
- Draw ROC curve and PR curve for object detection tasks, such as face detection.☆62Updated 7 years ago
- A Python 3 and Keras 2 implementation of MSCNN for people counting.☆68Updated 6 years ago
- tensorflow implementation for Yolo v4☆59Updated 4 years ago
- I use mobilenetv2 to export a model with 5 classes, then use libtorch to load and run it.☆18Updated 5 years ago
- Hybrid_Task_Cascade in mmdetection☆20Updated 6 years ago
- inference mask_rcnn model with tensorflow c++ api☆46Updated 6 years ago