sithu3's repositories
semantic-segmentation
SOTA Semantic Segmentation Models in PyTorch
PyTorch-ONNX-TFLite
Conversion of PyTorch Models into TFLite
sota-backbones
A collection of SOTA Image Classification Models in PyTorch
pose-estimation
Easy to use SOTA Top-Down Multi-person Pose Estimation Models in PyTorch
simple-object-tracking
Object Tracking with YOLOv5, CLIP, DINO and DeepSORT
PyTorch-ONNX-TRT
TensortRT installation and Conversion from PyTorch Models
audio-tagging
Easy to use Audio Tagging in PyTorch
self-supervised-learning
Fast, Simple and Easy to use SOTA Self-Supervised Image Classification Models in PyTorch
CUDA-Install-Guide
Installation guide for NVIDIA driver, CUDA, cuDNN and TensorRT
image-captioning
Simple and Easy to use Image Captioning Implementation
torch_optimize
Optimize PyTorch Models for faster inference
video-classification
Simple pipeline to get started with video classification and action recognition
Knowledge-Distillation-Pipeline
PyTorch Knowledge Distillation Framework
panoptic-segmentation
SOTA Panoptic Segmentation Models in PyTorch
magicnerf
A user-friendly and high-performance implementation of neural radiance fields (NeRF)
multimodal
SOTA Vision-Language Models for Text-to-Image, Image-to-Text, etc.
re_identification
Fast, Simple and Easy to configure Re-Identification Pipeline in PyTorch
RobustVideoMatting
Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!