mhubii / libtorch_custom_dataset
Demo dataset for libtorch
☆54Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for libtorch_custom_dataset
- A small example of using new PyTorch C++ frontend to implement ResNet☆43Updated 6 years ago
- Deep Learning sample programs using PyTorch in C++☆248Updated last year
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- Implementation maskrcnn-benchmark, pytorch c++ frontend☆81Updated 4 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆122Updated 5 years ago
- Libtorch C++ Examples☆52Updated 2 years ago
- PyTorch C++ API Documentation☆209Updated this week
- COCO API - Dataset @ http://cocodataset.org/☆55Updated 2 months ago
- Implementing CNN for Digit Recognition (MNIST and SVHN dataset) using PyTorch C++ API☆24Updated 2 years ago
- Parallel CUDA implementation of NON maximum Suppression☆79Updated 4 years ago
- convert torch module to tensorrt network or tvm function☆89Updated 4 years ago
- Some CUDA design patterns and a bit of template magic for CUDA☆146Updated last year
- Libtorch Examples☆41Updated 3 years ago
- CUDA implementation of NMS for PyTorch☆85Updated 4 years ago
- Example of how to use CUDA with CMake >= 3.8☆69Updated last year
- Event-Triggered Communication in Parallel Machine Learning☆25Updated 2 years ago
- Serving PyTorch 1.0 Models as a Web Server in C++☆226Updated 4 years ago
- AI-related samples made available by the DevTech ProViz team☆29Updated 7 months ago
- A sample for onnxparser working with trt user defined plugins for TRT7.0☆166Updated 4 years ago
- C++ API to log data in tensorboard format.☆76Updated 2 years ago
- A small framework to infer neural network☆141Updated this week
- Example code to create and train a Pytorch model using the new C++ frontend.☆17Updated 5 years ago
- YOLACT: Real-time Instance Segmentation on the FCOS detector (without bbox cropping), achives 35.2mAP on coco val☆90Updated 4 years ago
- ☆40Updated 8 months ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆83Updated 9 months ago
- Tencent NCNN with added CUDA support☆67Updated 3 years ago
- A simple pytorch implementation of CenterNet (CenterNet: Keypoint Triplets for Object Detection)☆31Updated 4 years ago
- ResNet Implementation, Training, and Inference Using LibTorch C++ API☆35Updated 5 months ago
- Count number of parameters / MACs / FLOPS for ONNX models.☆89Updated 3 weeks ago