Repository navigation

libtorch

Website
Wikipedia

pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

tensorrt libtorch 机器学习 PyTorch 深度学习 jetson Nvidia CUDA

Python

2727

362

3 小时前

prabhuomkar / pytorch-cpp

C++ Implementation of PyTorch Tutorials for Everyone

C++人工智能机器学习 PyTorch torch tensors 神经网络 autograd libtorch recurrent-neural-network 教程 Generative Adversarial Network convolutional-neural-network language-model colab datasets

C++

2034

266

1 个月前

dotnet / TorchSharp

A .NET library that provides access to the library that powers PyTorch.

libtorch mlnet torchscript

1579

195

2 天前

AllentDan / LibtorchTutorials

This is a code repository for pytorch c++ (or libtorch) tutorial.

libtorch classification segmentation C++unet vgg resnet yolov4 yolov4-tiny PyTorch tutorial-code

C++

780

129

3 年前

zhiqwang / yolort

yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.

libtorch yolov5 inference torchscript onnx onnxruntime tvm PyTorch detection jit visualization yolo lightning 部署 ncnn tensorrt nms

Python

729

151

19 天前

nanoporetech / dorado

Oxford Nanopore's Basecaller

libtorch nanopore genomics

C++

636

3 天前

Omegastick / pytorch-cpp-rl

PyTorch C++ Reinforcement Learning

PyTorch C++reinforcement-learning reinforcement-learning-algorithms a2c ppo pytorch-rl pytorch-cpp-frontend libtorch actor-critic advantage-actor-critic proximal-policy-optimization continuous-control

C++

522

5 年前

walktree / libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm

yolov3 libtorch PyTorch C++

C++

453

129

5 年前

AllentDan / LibtorchSegmentation

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.

semantic-segmentation fpn libtorch resnet resnext unet deeplabv3 deeplabv3plus C++pytorch-cpp-frontend models 神经网络 imagenet pretrained-weights PyTorch image-segmentation pspnet

C++

446

1 年前

hijkzzz / alpha-zero-gomoku

A Multi-threaded Implementation of AlphaZero (C++)

alphazero C++multithreading libtorch

Python

380

2 年前

yasenh / libtorch-yolov5

A LibTorch inference implementation of the yolov5

yolov5 libtorch gpu

C++

376

114

4 年前

BIGBALLON / PyTorch-CPP

PyTorch C++ inference with LibTorch

C++libtorch Demo inference PyTorch OpenCV imagenet

C++

329

4 年前

koba-jon / pytorch_cpp

Deep Learning sample programs using PyTorch in C++

PyTorch libtorch C++深度学习 Linux autoencoder vae dcgan pix2pix dimensionality-reduction image-to-image-translation semantic-segmentation anomaly-detection object-detection yolo

C++

250

2 年前

weixu000 / libtorch-yolov3-deepsort

yolov3 tracking imgui libtorch wxwidgets deepsort

C++

238

6 年前

LieluoboAi / radish

C++ model train&inference framework

深度学习 libtorch bert albert 自然语言处理

C++

224

5 年前

ttanzhiqiang / onnx_tensorrt_project

Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxnet to onnx to tensorrt

yolov4 yolov5 centernet yolor retinaface unet darknet libtorch PyTorch mxnet yolox

C++

213

4 年前