ros2_pytorch_cuda
by slabban
ROS2 PyTorch template node - running PyTorch C++ models in ROS2
AI summary
Model runner
Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment.
- stars
- 7
- forks
- 1
- watching
- 0
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
PyTorch node
A ROS2 node template to run PyTorch models in C++
MobileNetV2 model
An implementation of MobileNetV2 in PyTorch for image classification tasks.
CNN model
An implementation of a lightweight convolutional neural network architecture for mobile devices
Deep learning library
Provides a CUDA backend for the PyTorch deep learning framework
Model trainer
A PyTorch model fitting library designed to simplify the process of training deep learning models.
Robot model parser
Provides core data structure headers for describing robot models and interactions
Deep learning models
Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets.
Memory optimizer for deep learning models
A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution
Table layers
A collection of PyTorch implementations of Torch Table Layers to simplify the transition from old Torch architectures.
Model converter
Converts PyTorch models to their Torch equivalents.
ros/urdf63
Robot model parser
Provides tools and data structures for working with 3D models in robotics
Point Cloud Model
This is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks.
ResNet simulator
Reproduces ResNet-V3 with PyTorch for computer vision tasks
Data structure library
This project provides differentiable stacks and queues in PyTorch to facilitate the integration of data structures into models.
C++ PyTorch inference toolkit
A repository providing tools and examples to serve PyTorch models as C++ inference applications.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions ros2_pytorch_cuda.
