ros2_pytorch_cuda
Model runner
Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment.
ROS2 PyTorch template node - running PyTorch C++ models in ROS2
7 stars
0 watching
1 forks
Language: C++
last commit: over 2 years ago
Linked from 1 awesome list
Related projects:
Repository | Description | Stars |
---|---|---|
| A ROS2 node template to run PyTorch models in C++ | 39 |
| An implementation of MobileNetV2 in PyTorch for image classification tasks. | 271 |
| An implementation of a lightweight convolutional neural network architecture for mobile devices | 191 |
| Provides a CUDA backend for the PyTorch deep learning framework | 337 |
| A PyTorch model fitting library designed to simplify the process of training deep learning models. | 636 |
| Provides core data structure headers for describing robot models and interactions | 29 |
| Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |
| A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,823 |
| A collection of PyTorch implementations of Torch Table Layers to simplify the transition from old Torch architectures. | 89 |
| Converts PyTorch models to their Torch equivalents. | 541 |
| Provides tools and data structures for working with 3D models in robotics | 63 |
| This is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks. | 2,175 |
| Reproduces ResNet-V3 with PyTorch for computer vision tasks | 511 |
| This project provides differentiable stacks and queues in PyTorch to facilitate the integration of data structures into models. | 142 |
| A repository providing tools and examples to serve PyTorch models as C++ inference applications. | 226 |