zhangming8 / pytorch-cpp-model
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
- RetinaNet with backbone se_resnext50_32x4d, se_resnext101_32x4d☆24Updated 6 years ago
- 对https://github.com/ruinmessi/RFBNet 代码做中文注释,便于理解☆20Updated 6 years ago
- To augment image dataset for detection☆58Updated 6 years ago
- A simple demo to use TensorRT with pytorch☆28Updated 5 years ago
- 通过pytorch搭建卷积神经网络完成手写识别任务,并将训练好的模型以多种方式部署到TensorRT中加速。(1)通过加载权重。(2)通过加载onnx。(3)通过加载engine。☆44Updated 5 years ago
- 一个简单的TensorRT实现语义分割的C++程序☆59Updated 5 years ago
- Pytorch libtorch demo☆35Updated last year
- 深度学习/计算机视觉工程向笔记☆11Updated 5 years ago
- ☆15Updated 5 years ago
- 实现常用的one-stage和two-stage目标检测网络 华为媒体研究院 图文Caption、OCR识别、图视文多模态理解与生成相关方向工作或实习欢迎咨询 15757172165 https://guanfuchen.github.io/media/hw_zhaopin…☆55Updated 6 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
- cv的一些比赛☆46Updated 5 years ago
- TensorRT inference sample for ResNet50☆34Updated 6 years ago
- centerNet Caffe inference CPP☆69Updated 5 years ago
- CCFDF AI 数钢筋大赛☆172Updated 5 years ago
- I use mobilenetv2 to export a model with 5 classes, then use libtorch to load and run it.☆18Updated 4 years ago
- MobileNet v3☆48Updated 5 years ago
- Instance Segmentation, TensorMask☆51Updated 5 years ago
- GIoU loss simple example☆16Updated 6 years ago
- You can import this module directly☆55Updated 6 years ago
- Calling (TensorFlow) Python Program from C++☆51Updated 6 years ago
- ☆62Updated 5 years ago
- EfficientNetv2 TensorRT int8☆45Updated 4 years ago
- Summary of object detection(modules&&improvements)☆51Updated 5 years ago
- CenterNet's inference.(C++)/基于CenterNet的旋转目标检测C++版☆28Updated 4 years ago
- CenterNet with MobileNetV3 backboned helmet detection based on PyTorch with inference code only.☆52Updated 3 years ago
- mmdetection in python训练, libtorch in c++部署☆12Updated 4 years ago
- 津南数字制造季军解决方案☆36Updated 5 years ago
- A flexible Python front-end inference SDK based on TensorRT☆85Updated 3 years ago
- A bite of cpp api in PyTorch-1.0☆164Updated 5 years ago