mhubii / libtorch_custom_dataset
Demo dataset for libtorch
☆56Updated 2 years ago
Alternatives and similar repositories for libtorch_custom_dataset:
Users that are interested in libtorch_custom_dataset are comparing it to the libraries listed below
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆122Updated 6 years ago
- Example repository for custom C++/CUDA operators for TorchScript☆115Updated 2 years ago
- Implementation maskrcnn-benchmark, pytorch c++ frontend☆81Updated 4 years ago
- Libtorch C++ Examples☆53Updated 3 years ago
- Implementing CNN for Digit Recognition (MNIST and SVHN dataset) using PyTorch C++ API☆24Updated 2 years ago
- Example of how to use CUDA with CMake >= 3.8☆69Updated last year
- CUDA implementation of NMS for PyTorch☆85Updated 5 years ago
- C++ API to log data in tensorboard format.☆77Updated 2 months ago
- ResNet Implementation, Training, and Inference Using LibTorch C++ API☆39Updated 8 months ago
- Example code to create and train a Pytorch model using the new C++ frontend.☆17Updated 5 years ago
- convert torch module to tensorrt network or tvm function☆89Updated 5 years ago
- Deep Learning sample programs using PyTorch in C++☆250Updated last year
- An GPU/CUDA implementation of the Hungarian algorithm☆109Updated 5 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- A small example of using new PyTorch C++ frontend to implement ResNet☆43Updated 6 years ago
- Parallel CUDA implementation of NON maximum Suppression☆79Updated 4 years ago
- Event-Triggered Communication in Parallel Machine Learning☆25Updated 3 years ago
- Serving PyTorch 1.0 Models as a Web Server in C++☆226Updated 5 years ago
- Implementation of cv::Mat conversion to numpy.array for pybind11☆203Updated 3 years ago
- An unofficial Tensorflow implementation of the DSNT layer, as taken from the paper "Numerical Coordinate Regression with Convolutional Ne…☆64Updated 3 years ago
- A sample for onnxparser working with trt user defined plugins for TRT7.0☆166Updated 4 years ago
- Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …☆197Updated 5 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- Some CUDA design patterns and a bit of template magic for CUDA☆148Updated last year
- ☆22Updated 6 years ago
- ☆52Updated 4 years ago
- PyTorch C++ API Documentation☆216Updated this week
- Benchmark of TVM quantized model on CUDA☆111Updated 4 years ago
- ☆69Updated 5 years ago
- COCO API - Dataset @ http://cocodataset.org/☆55Updated 5 months ago