dusty-nv/pytorch-classification dusty-nv/pytorch-classification

  • Stars
    star
    28
  • Rank 882,216 (Top 18 %)
  • Language
    Python
  • Created over 5 years ago
  • Updated over 1 year ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
dusty-nv/pytorch-classification
github

dusty-nv

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Training of image classification models with PyTorch

More Repositories

1

jetson-inference

Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.
C++
7,694
star
2

jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
Python
2,017
star
3

jetson-reinforcement

Deep reinforcement learning GPU libraries for NVIDIA Jetson TX1/TX2 with PyTorch, OpenAI Gym, and Gazebo robotics simulator.
C++
875
star
4

ros_deep_learning

Deep learning inference nodes for ROS / ROS2 with support for NVIDIA Jetson and TensorRT
C++
870
star
5

jetson-utils

C++/CUDA/Python multimedia utilities for NVIDIA Jetson
C++
693
star
6

jetbot_ros

ROS nodes and Gazebo model for NVIDIA JetBot with Jetson Nano
Python
382
star
7

jetson-voice

ASR/NLP/TTS deep learning inference library for NVIDIA Jetson using PyTorch and TensorRT
Python
177
star
8

NanoLLM

Optimized local inference for LLMs with HuggingFace-like APIs for quantization, vision/language models, multimodal agents, speech, vector DB, and RAG.
Python
147
star
9

jetson-scripts

Configuration scripts and install guides for NVIDIA Jetson platform
Shell
96
star
10

turbo2

Tegra Unmanned Retrieval Bot gen2, with NVIDIA Jetson and deep reinforcement learning
C++
89
star
11

jetson-presentations

PDF slides about NVIDIA's Jetson embedded platform and deep learning.
59
star
12

camera-capture

GUI tool for collecting & labeling data from live camera feed
C++
41
star
13

pytorch-segmentation

Training of semantic segmentation networks with PyTorch
Python
35
star
14

rovernet

Deep reinforcement learning-driven robotic controller and navigation library using Torch.
C++
34
star
15

super-resolution

PyTorch super resolution model with RGB support and ONNX exporter
Python
31
star
16

pytorch-detection

Training of object detection networks with PyTorch
16
star
17

pytorch-odometry

Training of visual odometry estimation networks using PyTorch
Python
16
star
18

NanoDB

Zero-copy multimodal vector DB with CUDA and CLIP/SigLIP
Python
10
star
19

clip_trt

CLIP and SigLIP models optimized with TensorRT with a Transformers-like API
Python
8
star
20

gtc-llm-lab

Jetson AI Lab - LLM Inference
Shell
8
star
21

pytorch-timeseries

Python
7
star
22

ros2_pololu

ROS2 nodes for Pololu Maestro, Jrk, and Simple Motor Controller
C++
1
star