dusty-nv Goto Github PK

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-classification

Training of image classification models with PyTorch

pytorch-depth

MonoDepth Estimation - ICRA 2018 "Sparse-to-Dense: Depth Prediction from Sparse Depth Samples and a Single Image"

pytorch-detection

Training of object detection networks with PyTorch

pytorch-odometry

Training of visual odometry estimation networks using PyTorch

pytorch-segmentation

Training of semantic segmentation networks with PyTorch

pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

pytorch-timeseries

rl-baselines3-zoo

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

ros_deep_learning

Deep learning inference nodes for ROS / ROS2 with support for NVIDIA Jetson and TensorRT

rovernet

Deep reinforcement learning-driven robotic controller and navigation library using Torch.

small-stable-diffusion

stable-diffusion-webui-tensorrt

super-resolution

PyTorch super resolution model with RGB support and ONNX exporter

trt_pose

Real-time pose estimation accelerated with NVIDIA TensorRT

tts

🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production

turbo2

Tegra Unmanned Retrieval Bot gen2, with NVIDIA Jetson and deep reinforcement learning

vision

Datasets, Transforms and Models specific to Computer Vision

vision_msgs

Algorithm-agnostic computer vision message types for ROS.

zed-ros2-wrapper

ROS 2 wrapper beta for the ZED SDK